Matplotlib动画保存不会为曲面图创建动画

时间:2016-05-04 17:45:56

标签: python animation matplotlib

我正在尝试使用matplotlib.animation来动画表面的时间演变。找到了一个工作示例on this stackexchange question/answer。使用plt.show()我可以看到动画很好。问题是我试图保存它。当保存为gif或mp4时,我只从动画中获得一个。如果我正在进行1d动画,例如使用plt.plot(),我不会遇到此问题。以下是我的尝试:

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

def generate(X, Y, phi):
    R = 1 - np.sqrt(X**2 + Y**2)
    return np.cos(2 * np.pi * X + phi) * R

fig = plt.figure()
ax = axes3d.Axes3D(fig)

xs = np.linspace(-1, 1, 50)
ys = np.linspace(-1, 1, 50)
X, Y = np.meshgrid(xs, ys)
Z = generate(X, Y, 0.0)
wframe = ax.plot_surface(X, Y, Z, rstride=2, cstride=2, cmap=cm.coolwarm )
ax.set_zlim(-1,1)

def update(i, ax, fig):
    ax.cla()
    phi = i * 360 / 2 / np.pi / 100
    Z = generate(X, Y, phi)
    wframe = ax.plot_surface( X, Y, Z, rstride=2,
            cstride=2, cmap=cm.coolwarm )
    ax.set_zlim(-1,1)
    return wframe,

ani = animation.FuncAnimation( fig, update, frames=10,
        fargs=(ax, fig), interval=100 )

ani.save('plottest3d2.mp4', fps=30)
ani.save('plottest3d3.gif', fps=30, writer='imagemagick')

plt.show()

任何帮助解释显示情节和保存之间的差异都会很棒。

0 个答案:

没有答案