三维曲面(颜色贴图)

演示绘制使用coolwarm颜色贴图着色的3D表面。 使用antialiased = False使表面变得不透明。

还演示了使用LinearLocator和z轴刻度标签的自定义格式。

三维曲面(颜色贴图)示例

  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 import cm
  6. from matplotlib.ticker import LinearLocator, FormatStrFormatter
  7. import numpy as np
  10. fig = plt.figure()
  11. ax = fig.gca(projection='3d')
  13. # Make data.
  14. X = np.arange(-5, 5, 0.25)
  15. Y = np.arange(-5, 5, 0.25)
  16. X, Y = np.meshgrid(X, Y)
  17. R = np.sqrt(X**2 + Y**2)
  18. Z = np.sin(R)
  20. # Plot the surface.
  21. surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
  22.                        linewidth=0, antialiased=False)
  24. # Customize the z axis.
  25. ax.set_zlim(-1.01, 1.01)
  26. ax.zaxis.set_major_locator(LinearLocator(10))
  27. ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))
  29. # Add a color bar which maps values to colors.
  30. fig.colorbar(surf, shrink=0.5, aspect=5)
  32. plt.show()

下载这个示例