将matplotlib动画保存为mp4

92dk7w1h  于 2023-06-23  发布在  其他
关注(0)|答案(2)|浏览(136)

我想保存以下程序的输出动画在mp4中。该程序确实创建了一个mp4文件,但该文件是一个空白文件它不包含我想要的动画。我做错了什么?

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import matplotlib.animation as animation

plt.rcParams['animation.ffmpeg_path'] ='C:\\ffmpeg\\bin\\ffmpeg.exe'
fig=plt.figure()
ax=fig.add_subplot(111,projection="3d")

x=np.linspace(100,150,100)
t=(x-100)/0.5
y=-.01*np.cos(t)+.5*np.sin(t)+100.01
z=.01*np.sin(t)+.5*np.cos(t)+99.5

def animate(i):
    line.set_data(x[:i],y[:i])
    line.set_3d_properties(z[:i])

ax.set_xlim3d([min(x),max(x)])
ax.set_ylim3d([min(y),max(y)])
ax.set_zlim3d([min(z),max(z)])
ax.set_title("Particle in magnetic field") 
ax.set_xlabel("X")
ax.set_xlabel("Y")
ax.set_xlabel("Z")
line,=ax.plot([],[],[])
lin_ani=animation.FuncAnimation(fig,animate)
plt.legend()

FFwriter = animation.FFMpegWriter()
lin_ani.save('animation.mp4', writer = FFwriter, fps=10)
# plt.show()
ymzxtsji

ymzxtsji1#

当我学习起来知道,你应该这样写:

FFwriter = animation.FFMpegWriter(fps=10)
     lin_ani.save('animation.mp4', writer = FFwriter)

我是从这个网站1学到的

eqzww0vc

eqzww0vc2#

在函数的末尾添加此行

return line,

因此完整函数应该如下所示:

def animate(i):
    line.set_data(x[:i],y[:i])
    line.set_3d_properties(z[:i])
    return line,

相关问题