Python Matplotlib：指定图形大小时未释放内存

Question

我正在使用matplotlib生成许多数值模拟结果的图表。这些图用作视频中的帧，因此我通过重复调用与此函数类似的函数来生成许多图：

from pylab import *

def plot_density(filename,i,t,psi_Na):  
    figure(figsize=(8,6))
    imshow(abs(psi_Na)**2,origin = 'lower')
    savefig(filename + '_%04d.png'%i)
    clf()

问题在于，每次调用此函数时，python进程的内存使用量都会增加几兆字节。例如，如果我用以下循环调用它：

if __name__ == "__main__":
    x = linspace(-6e-6,6e-6,128,endpoint=False)
    y = linspace(-6e-6,6e-6,128,endpoint=False)
    X,Y = meshgrid(x,y)
    k = 1000000
    omega = 200
    times = linspace(0,100e-3,100,endpoint=False)
    for i,t in enumerate(times):
        psi_Na = sin(k*X-omega*t)
        plot_density('wavefunction',i,t,psi_Na)
        print i

然后ram的使用量会随着时间增长到600MB。但是，如果我figure(figsize=(8,6))在函数定义中注释掉该行，那么ram的使用将保持稳定在52MB。(8,6)是默认的图形尺寸，因此在两种情况下都会生成相同的图像。我想根据我的数值数据制作不同大小的图，而不会用完内存。如何强制python释放此内存？

我尝试了gc.collect()每个循环来强制进行垃圾收集，并且尝试f = gcf()获取当前数据然后del f将其删除，但无济于事。

我在64位Ubuntu 10.04上运行CPython 2.6.5。

Answer 1

从文档字符串中获取pylab.figure：

In [313]: pylab.figure?

如果要创建许多图形，请确保在未使用的图形上显式调用“关闭”，因为这将使pylab正确清理内存。

所以也许尝试：

pylab.close()     # closes the current figure