Python-用圖循環以保存matplotlib

Question

我有很多數字，我想將其保存到循環的每一步。 實際上是2位數字，如下所示：

plt.subplots_adjust(hspace=0.5)
plt.subplot(121)
plt.imshow(U2[40,:,:],cmap=cm.hot);plt.colorbar()

plt.subplot(122)
plt.imshow(V2[60,:,:],cmap=cm.hot);plt.colorbar()
plt.show()

我的問題是如何通過在每個步驟中將名稱更改為“ fig1”，“ fig2”作為示例來保存？ 我可以使用imsave嗎？

Answer 1

您可以將plt.show（）替換為plt.savefig（）並使用格式字符串（以確保數字以零填充為相同的長度）。 使用ffmpeg（或類似格式）從png文件制作視頻時，這會使生活變得更加輕松。 另外，值得稍加停頓以確保重繪該圖形。 對於您的情況，這看起來像

plt.subplots_adjust(hspace=0.5)
plt.subplot(121)
plt.imshow(U2[40,:,:],cmap=cm.hot);plt.colorbar()

plt.subplot(122)
plt.imshow(V2[60,:,:],cmap=cm.hot);plt.colorbar()

plt.pause(0.0001)
plt.savefig('./U2V2.{0:07d}.png'.format(step))

其中step是當前步驟號。 您還應該考慮為每個圖形（而不是步驟）使用一個序號，每次生成圖形時該序號只會改變一個（再次使視頻生成更容易）。