matplotlib 动画的内存使用情况

2024-02-22

我一直在调整此页面的代码http://jakevdp.github.io/blog/2012/08/18/matplotlib-animation-tutorial/ http://jakevdp.github.io/blog/2012/08/18/matplotlib-animation-tutorial/制作我自己的动画,但它很快就崩溃了。查看任务管理器,我可以看到运行程序所积累的内存在 30 秒内上升到 1GB,这对于我那台不太令人印象深刻的笔记本电脑来说非常重要。从代码每次调用animation(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()

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

anim = animation.FuncAnimation(fig, animate, init_func=init,
                               frames=200, interval=20, blit=True,
                               save_count=0)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

matplotlib 动画的内存使用情况 的相关文章

随机推荐