Как сохранить изображения в python в разных каталогах?

Question

Этот код делает изображения в цикле.Я хочу сохранить каждое изображение в своем собственном каталоге.Например, я хочу поместить img1 в папку img1, img2 в папку img2 и так далее.Я хочу использовать индекс в моем цикле в качестве суффикса.

 img = makeImage(g2_value,width=512)
        with open('%d.png' % i, 'wb') as f:
            f.write(img)

Answer 1

Лучше всего написать функцию, которая принимает fn в качестве ввода в качестве примера saveImg(fn, ...), а затем создать папку, если их не существует. Будьте очень осторожны, потому что вы можете переписать предыдущую работу.

import os
if os.path.exists("data") is False:
    os.mkdir("data")

fldr_template = "data/img{}"

for i in range(10):
    fldr = fldr_template.format(i)
    if os.path.exists(fldr) is False:
        os.mkdir(fldr)
    fn = "data/img{}/img{0}.png".format(i)
    # here you write your file to fn
    saveImg(fn, ...)

Answer 2

Используйте os.mkdir для создания каталога. Затем запишите изображение по правильному пути.

https://docs.python.org/2/library/os.html#os.mkdir

os.mkdir(str(i))
with open('%d/%d.png' % (i,i), 'wb') as f:
    f.write(img)