Python Matplotlib中使用plt.savefig存储图片的方法举例

前言

plt.show()展示图片的时候，截图进行保存，图片不是多么清晰

如何保存高清图也是一知识点

函数包名：import matplotlib.pyplot as plt

主要功能：

保存绘制数据后创建的图形。使用此方法可以将创建的图形保存

函数源码：（根据需要进行选择）

savefig(fname, dpi=None, facecolor='w', edgecolor='w', orientation='portrait', papertype=None,
format=None, transparent=False, bbox_inches=None, pad_inches=0.1, frameon=None, metadata=None)

参数解释：

正常保存：plt.savefig("xx.png")，也可以svg的格式进行保存

保存的时候需要plt.show()在plt.savefig()之后，顺序颠倒会出现图片为空白。

当前文件保存：

注意事项：

• 如果plt.show() 在plt.savefig()前，就会导致保存图片是空白的情况。

• window的路径读取，需要反斜杠

要把所有的参数用上，可以用在直方图上

import matplotlib.pyplot as plt

x =[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
plt.hist(x)

plt.savefig("squares1.png",
           bbox_inches ="tight",
           pad_inches = 1,
           transparent = True,
           facecolor ="g",
           edgecolor ='w',
           orientation ='landscape')

plt.show()

截图如下：

补充：解决plt.savefig() 保存多张图片有重叠的问题

问题描述：

在多次调用plt.savefig()时，出现了保存的图片有上一个数据出现并重叠的现象。如下图：

部分代码：

import matplotlib.pyplot as plt

def ch_graph(num_clusters, ch_score, filepath, method, module):
   # Plot ch graph
   plt.plot(num_clusters, ch_score, 'bx-')
   plt.xlabel('Number of cluster')
   plt.ylabel('Calinski-Harabasz Score')
   plt.title('Calinski-Harabasz Score against Number of Cluster')
   plt.grid(True)
filename = 'ch_graph_one.png'

folder = 'Picture/'
   ch_filepath = filepath + '/' + folder + filename
   plt.savefig(ch_filepath)

def elbow_graph(num_clusters, Sum_of_squared_distances, filepath, method, module):
   # Plot ch graph
   plt.plot(num_clusters, Sum_of_squared_distances, 'bx-')
   plt.xlabel('Number of cluster')
   plt.ylabel('Sum of squared dist')
   plt.title('Sum of squared dist against Number of Cluster')
   plt.grid(True)

filename = 'elbow_graph_one.png'
   folder = 'Picture/'
   elbow_filepath = filepath + '/' + folder + filename
   plt.savefig(elbow_filepath)

解决方法：

在plt.savefig()的下一行加上plt.close()就可以了。对于使用seaborn来绘制的图片，也同样使用plt.close()。

plt.close()内可输入的参数为：

1. None: 目前的figure

2. Figure: 给定的Figure实例

3. int: 一个 figure数

4. str: 一个 figure名字

5. ‘all’: 全部 figures

另外，有时候也会因为没有关闭上一个canvas, 导致出现以下问题：

fig.canvas.draw_idle()   # need this if 'transparent=True' to reset colors

来源：https://blog.csdn.net/weixin_47872288/article/details/128739356