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

您忘记了一个反斜杠：

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

注意：dir是一个内置函数-请勿这样命名您的变量。

Answer 2

使用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)

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

2 个答案: