三维曲面(棋盘格)

演示绘制以棋盘图案着色的3D表面。

三维曲面(棋盘格)示例

  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. from matplotlib.ticker import LinearLocator
  6. import numpy as np
  9. fig = plt.figure()
  10. ax = fig.gca(projection='3d')
  12. # Make data.
  13. X = np.arange(-5, 5, 0.25)
  14. xlen = len(X)
  15. Y = np.arange(-5, 5, 0.25)
  16. ylen = len(Y)
  17. X, Y = np.meshgrid(X, Y)
  18. R = np.sqrt(X**2 + Y**2)
  19. Z = np.sin(R)
  21. # Create an empty array of strings with the same shape as the meshgrid, and
  22. # populate it with two colors in a checkerboard pattern.
  23. colortuple = ('y', 'b')
  24. colors = np.empty(X.shape, dtype=str)
  25. for y in range(ylen):
  26.     for x in range(xlen):
  27.         colors[x, y] = colortuple[(x + y) % len(colortuple)]
  29. # Plot the surface with face colors taken from the array we made.
  30. surf = ax.plot_surface(X, Y, Z, facecolors=colors, linewidth=0)
  32. # Customize the z axis.
  33. ax.set_zlim(-1, 1)
  34. ax.w_zaxis.set_major_locator(LinearLocator(6))
  36. plt.show()

下载这个示例