如何向 Matplotlib 中的 imshow() 新增圖例？

要在 Matplotlib 中向 imshow() 新增圖例，我們可以執行以下步驟：

  • 設定圖形大小並調整各個子圖之間的和周圍的邊距。
  • 使用 numpy 建立隨機資料。
  • 初始化顏色圖。
  • 從第 2 步獲取樣本資料中的唯一資料點。
  • 繪製每種顏色，使用不同的標籤和顏色，以放在圖例上。
  • 在框內放置一個圖例，位於右上角。
  • 要顯示圖形，請使用 show() 方法。

示例

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

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

data = np.random.rand(3, 3)
cmap = cm.YlOrBr
unique_data = np.unique(data)
i = 0

for entry in unique_data:
   mycolor = cmap(entry * 255 / (max(unique_data) - min(unique_data)))
   plt.plot(0, 0, "-", color=mycolor, label="%d"%i)
   i += 1

plt.imshow(data, cmap=cmap)
plt.legend(loc="upper right", bbox_to_anchor=(1.25, 1.0))

plt.show()

輸出

更新於： 18-6-2021

4 千 + 次瀏覽

開啟你的 職業生涯

完成課程，獲得認證

開始
廣告