如何在 Python Matplotlib 中繪製密度圖？

要在 Python 中繪製密度圖，我們可以按照以下步驟操作——

  • 使用 numpy 建立 side、x、y 和 z。Numpy linspace 有助於根據第三個數在兩個點之間建立資料。

  • 使用 side 資料從座標向量返回座標矩陣。

  • 使用 x 和 y（步驟 2）建立指數資料。

  • 使用 pcolormesh() 方法建立具有非規則矩形網格的偽彩色圖。

  • 要顯示影像，請使用 show() 方法。

例項

from matplotlib import pyplot as plt, cm, colors
import numpy as np

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

side = np.linspace(-2, 2, 15)
X, Y = np.meshgrid(side, side)
Z = np.exp(-((X - 1) ** 2 + Y ** 2))

plt.pcolormesh(X, Y, Z, shading='auto')

plt.show()

輸出

更新於： 2021-9-29

3 千+ 瀏覽

開啟你的 職業生涯

完成課程獲得認證

開始學習
廣告