Matplotlib图形未保存在特定文件夹中

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

您忘记了一个反斜杠：

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

注意：dir是一个内置函数，不要这样命名变量

---