如何在 Python 中建立一個從綠色到紅色範圍的熱圖？（Matplotlib）

若要在 Python 中建立一個從綠色到紅色範圍的熱圖，我們可以採取以下步驟 -

  • 設定圖形大小並調整子圖之間及周圍的填充。
  • 為不同的顏色製作字典。
  • 使用 LinearSegmentedColormap 從線性對映片段中建立一個顏色圖。
  • 建立一個圖形和一組子圖。
  • 建立具有 5☓5 維度的隨機資料。
  • 建立一個具有不規則矩形網格的偽彩圖。
  • 為 ScalarMappable 例項 *mappable* 建立一個顏色條。
  • 要顯示圖形，請使用 show() 方法。

例子

import matplotlib.pyplot as plt
import matplotlib.colors as colors
import numpy as np

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

cdict = {'red': ((0.0, 0.0, 0.0),
                (0.5, 1.0, 1.0),
                (1.0, 1.0, 0.7)),

        'green': ((0.0, 0.7, 0.7),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0)),

         'blue': ((0.0, 0.0, 0.0),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0))
         }
GnRd = colors.LinearSegmentedColormap('GnRd', cdict)

fig, ax = plt.subplots(1)

data = np.random.rand(5, 5)*6.-3.

p = ax.pcolormesh(data, cmap=GnRd, vmin=-5, vmax=5)

fig.colorbar(p, ax=ax)

plt.show()

輸出

更新於： 07-07-2021

1K+ 瀏覽量

開啟您的職業生涯

完成課程以獲得認證

開始學習
廣告