matplotlib绘制大量图片内存问题

2024-01-25 16:37| 来源: 网络整理| 查看: 265

采用matplotlib绘制大量图片时会产生内存问题，最好的办法是，只创建一个 figure 对象，在画下一个图之前，使用 plt.clf() 清理掉 axes，这样可以复用 figure。

import matplotlib.pyplot as plt import numpy as np fig = plt.figure(figsize=(5,4)) # 在循环外部创建一个fig对象，循环利用 for i in range(10000): print('figure %d' % i) filename = 'D:\\Desktop\\计算机视觉\\data\\'+str(i)+'.txt' data = np.loadtxt(filename) x = data[:, 1] y = data[:, 0] plt.plot(x, y, 'k', linewidth=0.2) plt.axis('off') # 去除边框 plt.gca().xaxis.set_major_locator(plt.NullLocator()) plt.gca().yaxis.set_major_locator(plt.NullLocator()) plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, hspace = 0, wspace = 0) plt.margins(0,0) filesave = 'D:\\Desktop\\计算机视觉\\image\\'+str(i)+'.png' fig.savefig(filesave, format='png', transparent=True, dpi=200, pad_inches = 0) # 选择合适的分辨率 plt.clf() # 使用 plt.clf() 清理掉 axes

【本文地址】

公司简介

联系我们