Рисунок Matplotlib не сохраняется в указанной папке c

Question

Я пытаюсь сохранить фигуру matplotlib в r"C:\Users\USER\Handcrafted dataset\binary_image". но вместо сохранения рисунка в папке binary_image рисунок сохраняется в папке Handcrafted dataset. И имя изображения становится binary_image0.png. Но я хочу сохранить фигуру в желаемом каталоге как i.png. Как я могу это исправить?

di=r"C:\Users\USER\Handcrafted dataset\binary_image"
for i,img in enumerate(images):
    img = rgb2gray(img)
    plt.figure(figsize=(5,5))
    plt.imshow(img ,cmap='gray')
    plt.savefig(di+str(i)+".png")

Answer 1

Лучше использовать os.path.join или pathlib.Path.

import os

fn = "file_{}.png".format(i)
fn = os.path.join(dir, fn)
plt.savefig(fn)

или

from pathlib import Path

dir = Path(dir)
fn = dir / "file_{}.png".format(i)
plt.savefig(fn)

Answer 2

Вы забыли бэксла sh:

plt.savefig(save_to + '\' + str(i) + '.png')

Примечание. dir - встроенная функция - не называйте так свою переменную.

---