我正在尝试使用Matplotlib中的FuncAnimation类保存我创建的动画。我的动画更复杂,但是当我尝试保存给定here的简单示例时,我得到了同样的错误。
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.animation as animation
pause = False
def simData():
t_max = 10.0
dt = 0.05
x = 0.0
t = 0.0
while t < t_max:
if not pause:
x = np.sin(np.pi*t)
t = t + dt
yield x, t
def onClick(event):
global pause
pause ^= True
def simPoints(simData):
x, t = simData[0], simData[1]
time_text.set_text(time_template%(t))
line.set_data(t, x)
return line, time_text
fig = plt.figure()
ax = fig.add_subplot(111)
line, = ax.plot([], [], 'bo', ms=10) # I'm still not clear on this stucture...
ax.set_ylim(-1, 1)
ax.set_xlim(0, 10)
time_template = 'Time = %.1f s' # prints running simulation time
time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)
fig.canvas.mpl_connect('button_press_event', onClick)
ani = animation.FuncAnimation(fig, simPoints, simData, blit=False, interval=10,
repeat=True)
plt.show()
但是,当我尝试通过添加行
来保存此动画时ani.save('test.mp4')
最后,只保存前100帧。
保存动画后,该功能重新启动并按预期显示,显示和更新图形200次(或直到t达到t_max,无论我设置的是什么)。但保存的电影仅包含前100帧。
暂停功能使其变得棘手。没有它,我可以将帧= 200放入FuncAnimation调用,而不是使用我目前用于frame参数的迭代器/生成器类型函数。但是通过输入帧= 200,帧计数似乎是不可暂停的。
我该如何解决这个问题?
答案 0 :(得分:10)
ani = animation.FuncAnimation(fig, simPoints, simData, blit=False, interval=10,
repeat=True, save_count=200)
将解决这个问题。
在内部,save
仅保存固定数量的帧。如果传入固定长度的序列或数字,mpl可以正确地猜测长度。如果你传入一个(可能是无限的)生成器并且没有传入save_count
它默认为100。