我正在尝试在代码循环中打印一系列图像。我最终需要打印大约 1000 个来展示我的系统如何随时间变化。我已经查看了 Matplotlib runs out of memory when plotting in a loop 中概述的方法但我仍然无法让代码生成超过 96 张左右的图像。
我使用的剥离形式的代码如下
import numpy as np
import matplotlib as mpl
import os
def pltHM(graphname,graphtext,xAxis,yAxis,xMn,xMx,xCnt,yMn,yMx,yCnt,TCrt):
plt = mpl.pyplot
fig = plt.figure(figsize=(8,7), dpi=250)
cmap = mpl.cm.jet
norm = mpl.colors.Normalize(vmin=-3, vmax=3)
X = np.linspace(xMn,xMx,xCnt)
Y = np.linspace(yMn,yMx,yCnt)
plt.xlabel(xAxis)
plt.ylabel(yAxis)
plt.pcolormesh(X,Y,TCrt, cmap=cmap,norm=norm)
plt.grid(color='w')
plt.suptitle(graphtext, fontsize=14)
plt.colorbar()
plt.savefig(graphname, transparent = True)
plt.cla()
plt.clf()
plt.close(fig)
del plt
del fig
return
这在一个简单的循环中使用,如下所示
for loop1 in range(0,10):
for loop2 in range(0,100):
saveName = 'Test_Images/' + str(loop1) + '_' + str(loop2) + '.png'
plotHeatMap(saveName,'Test','X','Y',-35,35,141,-30,30,121,Z)
关于为什么上述内容没有释放内存并导致回溯消息的任何建议
运行时错误:无法为图像分配内存
非常感谢您提供的任何帮助
最佳答案
这是一个简单的示例,展示了您可以执行的操作。正如 Ajean 所指出的,您不应该像以前那样每次都导入 plt!一次就够了。另外,不要删除该图形并创建一个新图形...最好使用相同的图形并仅替换数据。
import numpy as np
import matplotlib.pyplot as plt
def plotHeatMap(fig, line, x, y, graphname):
line.set_data(x, y)
fig.canvas.draw()
fig.savefig(graphname)
fig1, ax1 = plt.subplots(1, 1)
line, = ax1.plot([],[])
ax1.set_xlim(0, 1)
ax1.set_ylim(0, 1)
for loop1 in range(0, 2):
for loop2 in range(0, 2):
x = np.random.random(100)
y = np.random.random(100)
save_name = 'fig_'+str(loop1) + '_' + str(loop2) + '.png'
plotHeatMap(fig1, line, x, y, save_name)
关于循环绘制图形时的 Python 内存,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38978920/