旋转3D线框图

一个非常简单的3D动画“动画”。 另请参见rotate_axes3d_demo。

（构建文档库时会跳过此示例，因为它有意运行需要很长时间）

  1. # This import registers the 3D projection, but is otherwise unused.
  2. from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused import
  4. import matplotlib.pyplot as plt
  5. import numpy as np
  6. import time
  9. def generate(X, Y, phi):
  10.     '''
  11.     Generates Z data for the points in the X, Y meshgrid and parameter phi.
  12.     '''
  13.     R = 1 - np.sqrt(X**2 + Y**2)
  14.     return np.cos(2 * np.pi * X + phi) * R
  17. fig = plt.figure()
  18. ax = fig.add_subplot(111, projection='3d')
  20. # Make the X, Y meshgrid.
  21. xs = np.linspace(-1, 1, 50)
  22. ys = np.linspace(-1, 1, 50)
  23. X, Y = np.meshgrid(xs, ys)
  25. # Set the z axis limits so they aren't recalculated each frame.
  26. ax.set_zlim(-1, 1)
  28. # Begin plotting.
  29. wframe = None
  30. tstart = time.time()
  31. for phi in np.linspace(0, 180. / np.pi, 100):
  32.     # If a line collection is already remove it before drawing.
  33.     if wframe:
  34.         ax.collections.remove(wframe)
  36.     # Plot the new wireframe and pause briefly before continuing.
  37.     Z = generate(X, Y, phi)
  38.     wframe = ax.plot_wireframe(X, Y, Z, rstride=2, cstride=2)
  39.     plt.pause(.001)
  41. print('Average FPS: %f' % (100 / (time.time() - tstart)))

下载这个示例