使用 Python Matplotlib 创建 3D 线框可视化

简介

本实验将指导你使用 Python 的 Matplotlib 库创建一个 3D 线框绘图。线框绘图是 3D 曲面的可视化表示，其中在曲面上的每个点之间绘制线条。这种类型的绘图对于可视化复杂的 3D 数据很有用，并且可以进行定制以创建令人印象深刻的可视化效果。

虚拟机提示

虚拟机启动完成后，点击左上角切换到“笔记本”标签以访问 Jupyter Notebook 进行练习。

有时，你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。

如果你在学习过程中遇到问题，请随时向 Labby 提问。课程结束后提供反馈，我们将立即为你解决问题。

导入库

第一步是导入必要的库。在本实验中，我们将使用 Matplotlib 库来创建线框绘图，并使用 NumPy 库来生成数据。

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d
import numpy as np

生成数据

接下来，我们将生成用于创建线框绘图的数据。在本实验中，我们将使用 np.meshgrid() 函数来创建 X、Y 和 Z 坐标。

## 生成数据
X, Y = np.meshgrid(np.arange(-5, 5, 0.25), np.arange(-5, 5, 0.25))
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

创建绘图

既然我们已经有了数据，就可以创建线框绘图了。在这个例子中，我们将使用 plot_wireframe() 函数来创建绘图。

## 创建绘图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(X, Y, Z, rstride=5, cstride=5)

自定义绘图

我们可以对绘图进行自定义，使其在视觉上更具吸引力。在这个例子中，我们将添加一个标题、坐标轴标签，并更改绘图的颜色。

## 自定义绘图
ax.set_title('线框绘图')
ax.set_xlabel('X 轴标签')
ax.set_ylabel('Y 轴标签')
ax.set_zlabel('Z 轴标签')
ax.plot_wireframe(X, Y, Z, rstride=5, cstride=5, color='green')

显示绘图

最后，我们可以使用 show() 函数来显示绘图。

## 显示绘图
plt.show()

总结

在本实验中，我们学习了如何使用 Python 的 Matplotlib 库创建三维线框绘图。我们使用 NumPy 生成数据，使用 plot_wireframe() 函数创建绘图，并对绘图进行自定义以使其在视觉上更具吸引力。我们还学习了如何添加标题、坐标轴标签以及更改绘图的颜色。有了这些知识，我们可以为复杂数据创建令人印象深刻的三维可视化效果。