用 Python 创建令人印象深刻的 3D 线框可视化

简介

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

虚拟机提示

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

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

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

Skills Graph

%%%%{init: {'theme':'neutral'}}%%%% flowchart RL python(("Python")) -.-> python/DataStructuresGroup(["Data Structures"]) matplotlib(("Matplotlib")) -.-> matplotlib/AdvancedPlottingGroup(["Advanced Plotting"]) python(("Python")) -.-> python/DataScienceandMachineLearningGroup(["Data Science and Machine Learning"]) matplotlib(("Matplotlib")) -.-> matplotlib/BasicConceptsGroup(["Basic Concepts"]) python(("Python")) -.-> python/BasicConceptsGroup(["Basic Concepts"]) python(("Python")) -.-> python/ModulesandPackagesGroup(["Modules and Packages"]) matplotlib/BasicConceptsGroup -.-> matplotlib/importing_matplotlib("Importing Matplotlib") matplotlib/BasicConceptsGroup -.-> matplotlib/figures_axes("Understanding Figures and Axes") python/BasicConceptsGroup -.-> python/comments("Comments") python/DataStructuresGroup -.-> python/tuples("Tuples") matplotlib/AdvancedPlottingGroup -.-> matplotlib/3d_plots("3D Plots") python/ModulesandPackagesGroup -.-> python/importing_modules("Importing Modules") python/ModulesandPackagesGroup -.-> python/using_packages("Using Packages") python/DataScienceandMachineLearningGroup -.-> python/numerical_computing("Numerical Computing") python/DataScienceandMachineLearningGroup -.-> python/data_visualization("Data Visualization") subgraph Lab Skills matplotlib/importing_matplotlib -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} matplotlib/figures_axes -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/comments -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/tuples -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} matplotlib/3d_plots -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/importing_modules -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/using_packages -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/numerical_computing -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} python/data_visualization -.-> lab-49034{{"使用 Python Matplotlib 创建 3D 线框可视化"}} end

导入库

第一步是导入必要的库。在本实验中，我们将使用 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() 函数创建绘图，并对绘图进行自定义以使其在视觉上更具吸引力。我们还学习了如何添加标题、坐标轴标签以及更改绘图的颜色。有了这些知识，我们可以为复杂数据创建令人印象深刻的三维可视化效果。