标签：注意 get error 解决 into nim int ann ctr
首先得安装Mayavi，看我之前的帖子，链接。
然后安装Moviepy，pip install moviepy. 关于Moviepy怎么用，可以见官网。
通过Mayavi绘制每一帧的图片，然后Moviepy获取到每一帧，生成视频文件。
1 import numpy as np 2 import mayavi.mlab as mlab 3 import moviepy.editor as mpy 4 5 duration= 2 # duration of the animation in seconds (it will loop) 6 7 # 使用MAYAVI先创建一个图片 8 fig_myv = mlab.figure(size=(220,220), bgcolor=(1,1,1)) 9 X, Y = np.linspace(-2,2,200), np.linspace(-2,2,200) 10 XX, YY = np.meshgrid(X,Y) 11 ZZ = lambda d: np.sinc(XX**2+YY**2)+np.sin(XX+d) 12 13 # 使用MoviePy把这个图片创建为一个动画，并保存 14 def make_frame(t): 15 mlab.clf() # clear the figure (to reset the colors) 16 mlab.mesh(YY,XX,ZZ(2*np.pi*t/duration), figure=fig_myv) 17 #下面两句很重要 18 f = mlab.gcf() 19 f.scene._lift() 20 return mlab.screenshot() #antialiased=True 21 22 animation = mpy.VideoClip(make_frame, duration=duration) 23 animation.write_gif("sinc.gif", fps=20)
值得注意的是，很多地方提供了类似于上面的代码，但是运行时会报错：“ValueError: cannot reshape array of size 12 into shape (0,0,3)”
解决办法是在 返回screenshot()前添加两行代码
1 f = mlab.gcf() 2 f.scene._lift()
运行结果如下
将我之前博客中的例子转换成动画
1 import mayavi.mlab as mlab 2 import numpy as np 3 from moviepy.editor import VideoClip 4 5 fig=mlab.figure(size=(200,200),bgcolor=(1.0,1.0,1.0)) 6 duration = 5 7 #生成20个点坐标 8 t = np.linspace(0, 4 * np.pi, 20) 9 x = np.sin(2 * t) 10 y = np.cos(t) 11 z = np.cos(2 * t) 12 s = 2 + np.sin(t) 13 14 def make_frame(t): 15 mlab.clf() 16 # # s数组只移动一位，点的颜色也就会变换 17 b = s[19] 18 s[1:20]=s[0:19] 19 s[0] = b 20 f = mlab.gcf() 21 f.scene._lift() 22 mlab.points3d(x, y, z, s, mode="cube",scale_mode="none", colormap="spectral", scale_factor=.5) 23 mlab.view(azimuth= 180*t/duration, distance=9) # camera angle 24 return mlab.screenshot(antialiased=True) 25 26 # duration是gif时长 27 animation = VideoClip(make_frame, duration=duration) 28 # fps帧率 29 animation.write_gif(‘rot.gif‘, fps=3)
本来小球的颜色是由s变量控制的，每一帧将s数组中的数平移一位，这样颜色点的颜色就能变化了。每一秒3帧，mlab.view控制相机变换。
运行结果如下
与Mayavi输出动画思路是一模一样的，区别在于返回每一帧图像的函数不同。
1 import matplotlib.pyplot as plt 2 import numpy as np 3 from moviepy.editor import VideoClip 4 from moviepy.video.io.bindings import mplfig_to_npimage 5 6 x = np.linspace(-2, 2, 200) 7 8 duration = 2 9 10 fig,ax = plt.subplots() 11 def make_frame(t): 12 ax.clear() 13 ax.plot(x, np.sin(x**2) + np.sin(x + 2*np.pi/duration * t), lw=3) 14 ax.set_ylim(-1.5, 2.5) 15 return mplfig_to_npimage(fig) 16 17 animation = VideoClip(make_frame, duration=duration) 18 animation.write_gif("matplotlib.gif", fps=4)
运行结果：
标签：注意 get error 解决 into nim int ann ctr
原文地址：https://www.cnblogs.com/dalanjing/p/12293627.html
