python - 使用相同的 python 脚本将两个图像保存到两个单独的文件中

Question

如何使用相同的 python 脚本将两个图像保存到两个单独的文件中。两个输出图像是相关矩阵和图形。我正在使用 matplotlib

imshow(matrix, interpolation='bilinear')
colorbar()
savefig('/home/sudipta/Downloads/phppython/'+ filename +'.png')
x = range(-width, width)
plt.plot(x, avg_vec)
plt.savefig('/home/sudipta/'+ filename +'plot' +'.png')

当我运行这个脚本时，这些数字相互重叠。但是，我需要两个单独的图像。最初，我试图将这些图像保存在同一目录中。我以为可能有问题。然后我试图保存在不同的目录中。但是，我没有成功

Answer 1

我认为您应该在两者之间添加 pyplot.clf() 。

#plot your first image
pyplot.savefig('filename.ext') # ext is your chosen filetype. jpg, png, etc.

# clear the plot
pyplot.clf()

# plot your second image
pyplot.savefig('filename2.ext')

函数 clf（清晰的图）从您正在绘制的当前轴中删除所有内容。

另一种选择是创建两个不同的图形对象。

# create figures and axes
fig0 = pyplot.figure()
ax0 = fig0.add_subplot(1, 1, 1)
fig1 = pyplot.figure()
ax1 = fig1.add_subplot(1, 1, 1)

# draw on ax0, e.g. with ax0.plot(...)
# draw on ax1

fig0.savefig('fig0.png')
fig1.savefig('fig1.png')

Answer 2

#plot your first image
pyplot.savefig('filename.ext') # ext is your chosen filetype. jpg, png, etc.

#plot your second image
pyplot.savefig('filename2.ext')