繪製帶有等值線圖投影的 Matplotlib 3D plot_surface


要繪製帶有等值線圖投影的 3D plot_surface,我們可以使用plot_surface()contourf()方法。

步驟

  • 設定圖形大小並調整子圖之間和周圍的填充。

  • 使用 numpy 建立x、y、X、YZ資料點。

  • 使用figure()方法建立一個新圖形或啟用現有圖形。

  • '~.axes.Axes'新增到圖形中作為子圖排列的一部分,並進行 3D 投影。

  • 使用plot_surface()方法建立曲面圖。

  • 使用contourf()方法建立 3D 填充等值線圖。

  • 關閉該座標軸。

  • 使用show()方法顯示圖形。

程式碼示例

import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.arange(-5, 5, 0.1)
y = np.arange(-5, 5, 0.1)
X, Y = np.meshgrid(x, y)
Z = X * np.exp(-X - Y)

fig = plt.figure()

ax = fig.add_subplot(111, projection='3d')
surf = ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=1, cmap="copper")
ax.contourf(X, Y, Z, zdir='z', offset=np.min(Z), cmap="plasma")
ax.contourf(X, Y, Z, zdir='x', offset=-5, cmap="PiYG_r")
ax.contourf(X, Y, Z, zdir='y', offset=5, cmap="PuBuGn")

ax.axis('off')
plt.show()

輸出

更新時間: 04-6 月 - 2021

652 views

開啟你的 職業生涯

完成課程以取得認證

開始
廣告
© . All rights reserved.