将情节保存到图像文件，而不是使用Matplotlib显示它

Question

我正在写一个快速而肮脏的脚本来动态生成情节。我使用以下代码（来自Matplotlib文档）作为起点：

from pylab import figure, axes, pie, title, show

# Make a square figure and axes
figure(1, figsize=(6, 6))
ax = axes([0.1, 0.1, 0.8, 0.8])

labels = 'Frogs', 'Hogs', 'Dogs', 'Logs'
fracs = [15, 30, 45, 10]

explode = (0, 0.05, 0, 0)
pie(fracs, explode=explode, labels=labels, autopct='%1.1f%%', shadow=True)
title('Raining Hogs and Dogs', bbox={'facecolor': '0.8', 'pad': 5})

show()  # Actually, don't show, just save to foo.png

我不想在GUI上显示绘图，而是想将绘图保存到一个文件（比如foo.png），以便可以在批处理脚本中使用。我该怎么做？

Answer 1

解决办法是：

pylab.savefig('foo.png')

Answer 2

正如其他人所说，plt。savefig（）或图1。savefig（）确实是保存图像的方法。

然而，我发现在某些情况下，图形总是显示出来。（例如，Spyder具有plt.ion（）：交互式模式=On。）我通过使用plt.close（figure_object）（请参阅留档）强制关闭巨型循环中的图形窗口来解决这个问题，因此在循环期间我没有一百万个打开的图形：

import matplotlib.pyplot as plt
fig, ax = plt.subplots( nrows=1, ncols=1 )  # create figure & 1 axis
ax.plot([0,1,2], [10,20,3])
fig.savefig('path/to/save/image/to.png')   # save the figure to file
plt.close(fig)    # close the figure window

Answer 3

虽然这个问题已经得到了回答，但我想在使用matplotlib时添加一些有用的提示。pyplot。保存图。文件格式可由扩展名指定：

from matplotlib import pyplot as plt

plt.savefig('foo.png')
plt.savefig('foo.pdf')

将分别给出光栅化或矢量化输出，这两种输出都很有用。此外，图像周围通常有一个不需要的空白，可以通过以下方式删除：

plt.savefig('foo.png', bbox_inches='tight')

请注意，如果显示绘图，plt。show（）应遵循plt。savefig（），否则文件图像将为空。