Matplotlib savefig() 在多个图上不断保存同一张图

Question

所以我有一个函数 generategraph(file)，它可以根据参数中的数据正确创建条形图，然后保存它。这是保存它的部分。

    plt.show()
    savefile = file.split('.txt')[0] + '.png'
    plt.savefig(savefile)

然后在 main 中，我将检查一组文件并在每个文件上调用 generategraph。

    for fil in files:
        generategraph(fil)

plt.show() 给了我正确的图表（每次不同的图表），但是当我转到保存的图表时，它们都是相同的图表（所以 len(files) 保存的图表数量，但每个都是第一个文件的图表，如果有意义的话）。我只是很困惑，因为 plt.show() 正在做我想让 plt.savefig 做的事情。

Answer 1

您正在使用状态机 (pyplot) 接口。别。

明确地创建你的人物：

fig1, ax1 = pyplot.subplots()

直接对它们采取行动：

lines, = ax1.plot(data1, data2, ...)

然后分别保存并关闭它们：

fig1.savefig(filename, dpi=300)
pyplot.close(fig1)

Answer 2

您可能需要验证您保存的图的名称是否不同。 （以下为伪代码，不清楚如何获取文件名。）

[edit]那么你可能应该把plt.show()放在plt.savefig(savefile)之后

#initialize idx to 0 earlier, and don't re-initialize it.

idx += 1
savefile = file + str(idx) + '.png'   # file might need to be replaced by a string
plt.savefig(savefile)
plt.show()              # place after plt.savefig()

或者，您更改​​generategraph的签名

def generategraph(file, idx):
    savefile = file + str(idx) + '.png'  # file might need to be replaced by a string
    plt.savefig(savefile)
    plt.show()              # place after plt.savefig()



for idx, fil in enumerate(files):
    generategraph(fil, idx)