如何在 Matplotlib 的 streamplot 中增加色圖/線寬質量?


要在 Matplotlib 的 streamplot 中增加色圖/線寬質量,我們可以採取以下步驟

步驟

  • 設定數字大小,並在子圖之間以及子圖周圍調整填充。

  • 建立圖形和一組子圖。

  • 建立xy 資料點,然後使用 np.meshgrid() 從座標向量返回座標矩陣。

  • 使用xy 資料點找到XY

  • 使用x, y, XY 資料點建立一個streamplot。可以使用該方法中的linewidth 引數增加線寬。此處我們使用 linewidth=5。

  • 為 ScalarMappable 例項建立顏色條,*stream.lines*

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

示例

import numpy as np
from matplotlib import pyplot as plt

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

fig, ax = plt.subplots()

x, y = np.meshgrid(np.linspace(-5, 5, 20), np.linspace(-5, 5, 20))

X = y
Y = 3 * x - 4 * y

stream = ax.streamplot(x, y, X, Y,  density=1, linewidth=5, cmap='plasma', color=Y)

fig.colorbar(stream.lines, ax=ax)

plt.show()

輸出

將產生以下輸出 −

更新日期:09-Oct-2021

222 次瀏覽

開啟您的 職業

完成該課程以獲得認證

開始學習
廣告
© . All rights reserved.