相同图中的2D和3D轴

此示例显示如何在同一图上绘制2D和3D绘图。

相同图中的2D和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. import numpy as np
  8. def f(t):
  9.     s1 = np.cos(2*np.pi*t)
  10.     e1 = np.exp(-t)
  11.     return np.multiply(s1, e1)
  14. # Set up a figure twice as tall as it is wide
  15. fig = plt.figure(figsize=plt.figaspect(2.))
  16. fig.suptitle('A tale of 2 subplots')
  18. # First subplot
  19. ax = fig.add_subplot(2, 1, 1)
  21. t1 = np.arange(0.0, 5.0, 0.1)
  22. t2 = np.arange(0.0, 5.0, 0.02)
  23. t3 = np.arange(0.0, 2.0, 0.01)
  25. ax.plot(t1, f(t1), 'bo',
  26.         t2, f(t2), 'k--', markerfacecolor='green')
  27. ax.grid(True)
  28. ax.set_ylabel('Damped oscillation')
  30. # Second subplot
  31. ax = fig.add_subplot(2, 1, 2, projection='3d')
  33. X = np.arange(-5, 5, 0.25)
  34. Y = np.arange(-5, 5, 0.25)
  35. X, Y = np.meshgrid(X, Y)
  36. R = np.sqrt(X**2 + Y**2)
  37. Z = np.sin(R)
  39. surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1,
  40.                        linewidth=0, antialiased=False)
  41. ax.set_zlim(-1, 1)
  43. plt.show()

下载这个示例