【Python】解决使用plt.savefig保存图片时一片空白

原文链接： http://blog.csdn.net/u010099080/article/details/52912439

问题

当使用如下代码保存使用plt.savefig 保存生成的图片时，结果打开生成的图片确实一片空白。

import matplotlib.pyplot as plt

""" 一些画图代码 """

plt.show()
plt.savefig("filename.png")

原因

其实产生这个现象的原因很简单：在plt.show() 后调用了plt.savefig() ，在plt.show() 后实际上已经创建了一个新的空白的图片（座标轴），这时候你再plt.savefig() 就会保存这个新生成的空白图片。

解决

知道了原因，就不难知道解决办法了，解决办法有两种：

1、在plt.show() 之前调用plt.savefig()；

import matplotlib.pyplot as plt

""" 一些画图代码 """

plt.savefig("filename.png")
plt.show()

2、画图的时候获取当前图像（这一点非常类似于Matlab的句柄的概念）

import matplotlib.pyplot as plt

""" 一些画图代码 """

plt.savefig("filename.png")
plt.show()