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)

有关上述原因未释放内存并导致回溯消息的任何建议

RuntimeError：无法为图像分配内存

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

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)

在循环中绘制图形时的Python内存

1 个答案: