matplotlib动画的内存使用情况

Question

我一直在调整这个页面的代码http://jakevdp.github.io/blog/2012/08/18/matplotlib-animation-tutorial/来制作我自己的动画，但它很快崩溃。看看任务管理器，我可以看到运行程序的内存增加到30秒内达到1GB，这对于我不太令人印象深刻的笔记本电脑来说非常重要。从代码调用动画的方式（i）每次都在线上设置y_data，旧数据是否未被替换，导致内存增加？我想解决这个问题。我关于matplotlib内部工作的知识是有限的，我尝试过的一些事情是在动画（i）中放置close（），clf（）和gc.collect（），但它们都不起作用。

""" 
Matplotlib Animation Example 

author: Jake Vanderplas 
email: vanderplas@astro.washington.edu 
website: http://jakevdp.github.com 
license: BSD 
Please feel free to use and modify this, but keep the above information. Thanks! 
""" 

import numpy as np 
from matplotlib import pyplot as plt 
from matplotlib import animation 

# First set up the figure, the axis, and the plot element we want to animate 
fig = plt.figure() 
ax = plt.axes(xlim=(0, 2), ylim=(-2, 2)) 
line, = ax.plot([], [], lw=2) 

# initialization function: plot the background of each frame 
def init(): 
    line.set_data([], []) 
    return line, 

# animation function. This is called sequentially 
def animate(i): 
    x = np.linspace(0, 2, 1000) 
    y = np.sin(2 * np.pi * (x - 0.01 * i)) 
    line.set_data(x, y) 
    return line, 

# call the animator. blit=True means only re-draw the parts that have changed. 
anim = animation.FuncAnimation(fig, animate, init_func=init, 
           frames=200, interval=20, blit=True) 

# save the animation as an mp4. This requires ffmpeg or mencoder to be 
# installed. The extra_args ensure that the x264 codec is used, so that 
# the video can be embedded in html5. You may need to adjust this for 
# your system: for more information, see 
# http://matplotlib.sourceforge.net/api/animation_api.html 
# anim.save('basic_animation.mp4', fps=30, extra_args=['-vcodec', 'libx264']) 

plt.show() 

Answer 1

默认情况下，动画代码会尝试保存一些帧数（默认值为100）。尝试明确地将其设置为零：

anim = animation.FuncAnimation(fig, animate, init_func=init, 
           frames=200, interval=20, blit=True, 
           save_count=0)