我正在编写一个快速而肮脏的脚本来动态生成绘图。我使用下面的代码(来自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),因此,例如,它可以在批处理脚本中使用。我怎么做?
虽然问题已得到解答,但我想在使用matplotlib.pyplot.savefig时添加一些有用的提示。文件格式可以通过扩展名指定:
from matplotlib import pyplot as plt
plt.savefig('foo.png')
plt.savefig('foo.pdf')
将分别给出栅格化或矢量化输出,两者都可能有用。此外,你会发现pylab
在图像周围留下了一个慷慨的,通常是不受欢迎的空白空间。删除它:
savefig('foo.png', bbox_inches='tight')
解决方案是:
pylab.savefig('foo.png')
正如其他人所说, plt.savefig()
或fig1.savefig()
确实是保存图像的方法。
但是我发现在某些情况下(例如,Spyder 有plt.ion()
:interactive mode = On),总会显示数字。我通过在巨型循环中强制关闭图形窗口来解决这个问题,所以在循环期间我没有一百万个开放数字:
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