3D绘图作为子图

展示包括3D图作为子图。

3D绘图作为子图示例

  1. import matplotlib.pyplot as plt
  2. from matplotlib import cm
  3. import numpy as np
  5. from mpl_toolkits.mplot3d.axes3d import get_test_data
  6. # This import registers the 3D projection, but is otherwise unused.
  7. from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused import
  10. # set up a figure twice as wide as it is tall
  11. fig = plt.figure(figsize=plt.figaspect(0.5))
  13. #===============
  14. #  First subplot
  15. #===============
  16. # set up the axes for the first plot
  17. ax = fig.add_subplot(1, 2, 1, projection='3d')
  19. # plot a 3D surface like in the example mplot3d/surface3d_demo
  20. X = np.arange(-5, 5, 0.25)
  21. Y = np.arange(-5, 5, 0.25)
  22. X, Y = np.meshgrid(X, Y)
  23. R = np.sqrt(X**2 + Y**2)
  24. Z = np.sin(R)
  25. surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
  26.                        linewidth=0, antialiased=False)
  27. ax.set_zlim(-1.01, 1.01)
  28. fig.colorbar(surf, shrink=0.5, aspect=10)
  30. #===============
  31. # Second subplot
  32. #===============
  33. # set up the axes for the second plot
  34. ax = fig.add_subplot(1, 2, 2, projection='3d')
  36. # plot a 3D wireframe like in the example mplot3d/wire3d_demo
  37. X, Y, Z = get_test_data(0.05)
  38. ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
  40. plt.show()

下载这个示例