要實現3D曲面圖,可以使用Matplotlib中的mplot3d模塊。以下是一個簡單的示例代碼來繪制一個3D曲面圖:
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
# 生成數據
X = np.linspace(-5, 5, 100)
Y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 創建一個3D圖形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 繪制3D曲面圖
surf = ax.plot_surface(X, Y, Z, cmap='viridis')
# 添加顏色欄
fig.colorbar(surf)
# 顯示圖形
plt.show()
在這個示例中,我們生成了一個二維網格數據,并計算了一個曲面函數。然后創建了一個3D圖形,并使用plot_surface()方法繪制了曲面圖。最后添加了一個顏色欄來表示Z值的變化。運行這段代碼將繪制一個簡單的3D曲面圖。