18 Sep 2024

如何使用 Python Matplotlib 绘制 3D 曲面图

在数据可视化中，3D 图表是一个非常有用的工具，特别是当我们想要展示复杂的三维数据时，如期权的波动率曲面。Python 的 matplotlib 库提供了生成各种类型图表，包括 3D 图表。

本文将介绍如何使用 Python 中的 matplotlib 绘制 3D 曲面图，适用于不同领域的数据可视化需求。

准备工作

安装 matplotlib，命令如下：

1
pip install matplotlib

绘制简单的 3D 曲面图

引入所需库：为了绘制 3D 图形，我们需要使用 matplotlib 中的 Axes3D 和 plot_surface 方法。为了演示，还要引入 numpy 生成绘图数据。

1
2
3
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D # 显式导入Axes3D，确保版本兼容

生成演示数据：3D 曲面图通常是由一个三维网格点组成的，其中 X 轴和 Y 轴分别代表行和列，Z 轴表示每个网格点的高度值。我们可以使用 numpy 来生成 X 和 Y 轴的网格，同时基于 X 和 Y 生成 Z 的值。

1
2
3
4
5
6
7
8
9
# 使用 numpy 生成 X 和 Y 的数据
x = np.linspace(-5, 5, 100)  # 生成从 -5 到 5 的 100 个等间距的点
y = np.linspace(-5, 5, 100)  # 同样为 Y 轴生成相同范围的点

# 生成二维网格
X, Y = np.meshgrid(x, y)

# 定义 Z 轴数据，使用一个简单的函数 Z = f(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))

在这个例子中，Z 轴数据是 X 和 Y 的平方和的平方根的正弦值，我将使用这个数据绘制曲面。

绘制 3D 曲面图

接下来使用 matplotlib 的 plot_surface 方法来绘制曲面。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
# 创建 3D 图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制 3D 曲面图
surf = ax.plot_surface(X, Y, Z, cmap='viridis')

# 为图表添加颜色条
fig.colorbar(surf)

# 设置坐标轴标签
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

# 显示图形
plt.show()

运行此代码后，您将看到一个 3D 曲面图。

了解了基本的 3D 曲面图绘制后，接下来开始探讨一些更高级的特性，如自定义颜色、设置透明度、添加线框等。

添加线框和透明度

有时，在 3D 曲面图上添加线框或调整透明度可以帮助我们更好地理解数据结构。以下代码展示了如何添加这些特性。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
# 创建 3D 图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制带线框的 3D 曲面图，alpha 用于设置透明度
surf = ax.plot_surface(X, Y, Z, cmap='plasma', edgecolor='none', alpha=0.8)

# 添加网格线框（wireframe）
ax.plot_wireframe(X, Y, Z, color='black', linewidth=0.5)

# 为图表添加颜色条
fig.colorbar(surf)

# 设置坐标轴标签
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

# 显示图形
plt.show()

示例中，通过 plot_wireframe() 添加了网格线框，颜色映射使用了 plasma，通过 alpha=0.8 设置了透明度为 80%。

效果如下所示：

控制图形视角

matplotlib 提供了对 3D 图形视角的控制。可以通过 ax.view_init() 来设置视角（即观察图形的角度），elev 参数设置仰角，azim 参数设置方位角。

1
2
# 设置 60 仰角和 45 方位角
ax.view_init(elev=60, azim=45)

通过调整这些参数，您可以从不同的角度观察 3D 曲面图。

Matplotlib 中绘制 3D 曲面图要点

创建数据网格：使用 numpy.meshgrid 生成二维的 X 和 Y 网格，并根据需要定义 Z 轴的值。
绘制曲面图：使用 matplotlib 的 plot_surface() 方法来绘制 3D 曲面，使用 cmap 来调整颜色映射。
自定义图形：可以添加透明度、线框，或者通过自定义函数来生成 Z 轴数据。同时，还可以通过 view_init() 调整视角。
可视化增强：为图形添加颜色条，调整坐标轴标签，使用不同的颜色映射函数来使数据更加清晰。

总结

使用 matplotlib 绘制 3D 曲面图帮助我们可视化复杂的三维数据。通过掌握基础的网格生成和绘图函数，以及对图形的进一步自定义和优化，就可轻松创建适合您需求的 3D 可视化图表。

本文介绍了从基础的 3D 曲面图绘制方法，希望对你有所帮助。

本文来源于 POLOXUE's BLOG，地址：如何使用 Python Matplotlib 绘制 3D 曲面图

欢迎关注我的公众号：

POLOXUE's BLOG