matplotlib动画的内存使用情况

时间:2013-10-28 22:50:50

标签: python animation matplotlib

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

"""
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()

1 个答案:

答案 0 :(得分:2)

默认情况下,动画代码会尝试保存一定数量的帧(默认为100)。尝试将其明确设置为零:

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