如何在保存matplotlib动画到文件时指定持续时间?通常它是由animation.FuncAnimation()
的frame
参数给出的,但当使用生成器创建动画帧时就不是这样了。例如,使用
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
def simData():
t_max = 10.0
dt = 0.05
x = 0.0
t = 0.0
while t < t_max:
x = np.sin(np.pi*t)
t = t + dt
yield x, t
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)
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)
ani = animation.FuncAnimation(fig, simPoints, simData)
#plt.show()
ani.save('animation.mp4', writer="avconv", codec="libx264")
创建一个20秒的视频显示大约。5秒的"模拟时间",用plt.show()
显示时生成的帧数的一半。
您在FuncAnimation
上缺少save_count
关键字。如果你传入一个生成器,那么你可以传入要保存的帧数:
ani = animation.FuncAnimation(fig, simPoints, simData, save_count=200)
迭代似乎会继续下去,直到生成器耗尽或达到save_count
。默认值是100,尽管在源代码之外似乎没有很清楚地说明。