用matplotlib保存动画?

时间:2014-05-21 19:16:36

标签: python animation matplotlib

我已经使用自制软件安装了ffmpeg。

然而,我现在面临的问题是在运行脚本时没有保存动画。 我已经在我的脚本和我的.bash_profile

下面添加了
import matplotlib
matplotlib.use('TKAgg')
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)

的.bash_profile

export PATH=$PATH:/$HOME/../../usr/local/Celler/ffmpreg/2.2.2/bin
export FFMPEG_BIN=/$HOME/../../usr/local/Cellar/ffmpeg/2.2.2/bin/

1 个答案:

答案 0 :(得分:0)

通过使用带有Anaconda环境的ipython笔记本,我能够正确输出动画。

需要一些工作来让动画在ipython笔记本中显示。这是:

from matplotlib import animation 
from tempfile import NamedTemporaryFile

VIDEO_TAG = """<video controls>
 <source src="data:video/x-webm;base64,{0}" type="video/webm">
 Your browser does not support the video tag.
</video>"""

def anim_to_html(anim):
 if not hasattr(anim, '_encoded_video'):
    with NamedTemporaryFile(suffix='.webm') as f:
        anim.save(f.name, fps=6, extra_args=['-vcodec', 'libvpx'])
        video = open(f.name, "rb").read()
    anim._encoded_video = video.encode("base64")

return VIDEO_TAG.format(anim._encoded_video)

from IPython.display import HTML

def display_animation(anim):
  plt.close(anim._fig)
  return HTML(anim_to_html(anim))

完成上述代码后,以下动画正确输出到ipython笔记本中:

# 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)

anim.save('basic_animation.mp4', fps=30, extra_args=['-vcodec', 'libx264'])

display_animation(anim)

由于某些原因,当我右键单击并保存时,我无法以可读格式查看电影...如果有人知道或找出解决方案,请告诉我!