如何在for循环中保存多个绘图

Question

我正试图在for循环中保存多个绘图，但我收到了以下错误：'FileNotFoundError:[Erno 2]没有这样的文件或目录：'8/2020/graph.png'

这是我的代码：

for i in names.columns:
fig, ax = plt.subplots(1,1,figsize=(150,50))    
g1 = sns.heatmap(df, cmap="Spectral", 
cbar_kws={'shrink': 0.85}, 
)
# plt.savefig(i + 'graph.png')
fig.savefig('{}/graph.png'.format(i))
plt.show()

但当我使用--->plt.savefig('seaborn_change_colormap_size_02.png'(我的代码可以工作，但问题是最后我只有一个情节(最后一个(

有什么想法可以用不同的名字保存所有的情节吗？

Answer 1

如果目录还不存在，则必须创建该目录。

import os
for i in names.columns:
fig, ax = plt.subplots(1,1,figsize=(150,50))    
g1 = sns.heatmap(df, cmap="Spectral", 
cbar_kws={'shrink': 0.85}, 
)
if not os.path.isdir(i):
os.mkdir(i)
fig.savefig(f'{i}/graph.png')