1.0版本新特性:3d子图

在同一图中创建两个三维图。

  1. # This import registers the 3D projection, but is otherwise unused.
  2. from mpl_toolkits.mplot3d import Axes3D # noqa: F401 unused import
  3. from matplotlib import cm
  4. #from matplotlib.ticker import LinearLocator, FixedLocator, FormatStrFormatter
  5. import matplotlib.pyplot as plt
  6. import numpy as np
  7. fig = plt.figure()
  8. ax = fig.add_subplot(1, 2, 1, projection='3d')
  9. X = np.arange(-5, 5, 0.25)
  10. Y = np.arange(-5, 5, 0.25)
  11. X, Y = np.meshgrid(X, Y)
  12. R = np.sqrt(X**2 + Y**2)
  13. Z = np.sin(R)
  14. surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet,
  15. linewidth=0, antialiased=False)
  16. ax.set_zlim3d(-1.01, 1.01)
  17. #ax.w_zaxis.set_major_locator(LinearLocator(10))
  18. #ax.w_zaxis.set_major_formatter(FormatStrFormatter('%.03f'))
  19. fig.colorbar(surf, shrink=0.5, aspect=5)
  20. from mpl_toolkits.mplot3d.axes3d import get_test_data
  21. ax = fig.add_subplot(1, 2, 2, projection='3d')
  22. X, Y, Z = get_test_data(0.05)
  23. ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
  24. plt.show()

3d子图示例

参考

此示例中显示了以下函数,方法,类和模块的使用:

  1. import matplotlib
  2. import mpl_toolkits
  3. matplotlib.figure.Figure.add_subplot
  4. mpl_toolkits.mplot3d.axes3d.Axes3D.plot_surface
  5. mpl_toolkits.mplot3d.axes3d.Axes3D.plot_wireframe
  6. mpl_toolkits.mplot3d.axes3d.Axes3D.set_zlim3d

下载这个示例