循环绘制图形时的 Python 内存

Question

我正在尝试在代码循环中打印一系列图像。我最终需要打印大约 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)

关于为什么上述内容没有释放内存并导致回溯消息的任何建议

运行时错误:无法为图像分配内存

非常感谢您提供的任何帮助

Answer 1

这是一个简单的示例，展示了您可以执行的操作。正如 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)