Python使用Matplotlib繪制3D曲面圖詳解

更新時間：2025年05月01日 12:19:16 作者：波羅學

這篇文章主要介紹了Python使用Matplotlib繪制3D曲面圖,在Python中,使用Matplotlib庫繪制3D曲面圖可以通過mpl_toolkits.mplot3d模塊實現(xiàn),需要的朋友可以參考下

準備工作

安裝 matplotlib，命令如下：

pip install matplotlib

繪制簡單的 3D 曲面圖

引入所需庫：為了繪制 3D 圖形，我們需要使用 matplotlib 中的 Axes3D 和 plot_surface 方法。為了演示，還要引入 numpy 生成繪圖數(shù)據(jù)。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D # 顯式導入Axes3D，確保版本兼容

生成演示數(shù)據(jù)：3D 曲面圖通常是由一個三維網(wǎng)格點組成的，其中 X 軸和 Y 軸分別代表行和列，Z 軸表示每個網(wǎng)格點的高度值。我們可以使用 numpy 來生成 X 和 Y 軸的網(wǎng)格，同時基于 X 和 Y 生成 Z 的值。

# 使用 numpy 生成 X 和 Y 的數(shù)據(jù)
x = np.linspace(-5, 5, 100)  # 生成從 -5 到 5 的 100 個等間距的點
y = np.linspace(-5, 5, 100)  # 同樣為 Y 軸生成相同范圍的點
# 生成二維網(wǎng)格
X, Y = np.meshgrid(x, y)
# 定義 Z 軸數(shù)據(jù)，使用一個簡單的函數(shù) Z = f(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))

在這個例子中，Z 軸數(shù)據(jù)是 X 和 Y 的平方和的平方根的正弦值，我將使用這個數(shù)據(jù)繪制曲面。

繪制 3D 曲面圖

接下來使用 matplotlib 的 plot_surface 方法來繪制曲面。

# 創(chuàng)建 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 曲面圖上添加線框或調(diào)整透明度可以幫助我們更好地理解數(shù)據(jù)結(jié)構(gòu)。以下代碼展示了如何添加這些特性。

# 創(chuàng)建 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)
# 添加網(wǎng)格線框（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() 添加了網(wǎng)格線框，顏色映射使用了 plasma，通過 alpha=0.8 設置了透明度為 80%。

效果如下所示：

控制圖形視角

matplotlib 提供了對 3D 圖形視角的控制。可以通過 ax.view_init() 來設置視角（即觀察圖形的角度），elev 參數(shù)設置仰角，azim 參數(shù)設置方位角。

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

通過調(diào)整這些參數(shù)，您可以從不同的角度觀察 3D 曲面圖。

Matplotlib 中繪制 3D 曲面圖要點

創(chuàng)建數(shù)據(jù)網(wǎng)格：使用 numpy.meshgrid 生成二維的 X 和 Y 網(wǎng)格，并根據(jù)需要定義 Z 軸的值。
繪制曲面圖：使用 matplotlib 的 plot_surface() 方法來繪制 3D 曲面，使用 cmap 來調(diào)整顏色映射。
自定義圖形：可以添加透明度、線框，或者通過自定義函數(shù)來生成 Z 軸數(shù)據(jù)。同時，還可以通過 view_init() 調(diào)整視角。
可視化增強：為圖形添加顏色條，調(diào)整坐標軸標簽，使用不同的顏色映射函數(shù)來使數(shù)據(jù)更加清晰。