解决使用 plt.savefig 保存图片时一片空白[python]

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

import matplotlib.pyplot as plt

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

plt.show()
plt.savefig("../混淆矩阵.png")

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

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

import matplotlib.pyplot as plt

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

plt.savefig("../混淆矩阵.png")
plt.show()

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

    # gcf: Get Current Figure
    fig = plt.gcf()
    plt.show()
    fig.savefig("../混淆矩阵.png")