如何向 matplotlib 图形添加 3D 子图?\

matplotlibpythondata visualization

要向 matplotlib 图形添加 3D 子图,我们可以采取以下步骤 −

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用 numpy 创建 x、y 和 z 数据点。
  • 创建新图形或激活现有图形。
  • 添加 'ax'以 projection='3d' 的形式将 x、y 和 z 数据点添加到图中。
  • 使用 plot() 方法绘制 x、y 和 z 数据点。
  • 要显示该图,请使用 .show() 方法。

示例

from matplotlib import pyplot as plt
import numpy as np

# 设置图大小
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

# 使用 numpy 创建 x、y 和 z 数据点
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
z = x ** 2 + y ** 2

fig = plt.figure()

# 添加子图,投影为"3d"
ax = fig.add_subplot(111, project="3d")

# 绘制 x、y 和 z 数据点
ax.plot(x, y, z, color='red', lw=7)

plt.show()

输出

它将产生以下输出 −


相关文章