如何在 Matplotlib 中顯示矩陣值和顏色對映?


要在 Matplotlib 中顯示矩陣值和顏色對映,我們可以採取以下步驟:

  • 設定影像大小並調整子圖之間的和周圍的邊距。
  • 建立一個影像和一組子圖。
  • 初始化矩陣的最大和最小值。
  • 將 2D 矩陣或陣列的值繪製為顏色編碼的影像。
  • 遍歷顏色編碼影像的每個單元,並將值置於中心。
  • 要顯示影像,請使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

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

fig, ax = plt.subplots()
min_val, max_val = 0, 5
matrix = np.random.randint(0, 5, size=(max_val, max_val))
ax.matshow(matrix, cmap='ocean')

for i in range(max_val):
   for j in range(max_val):
      c = matrix[j, i]
      ax.text(i, j, str(c), va='center', ha='center')

plt.show()

輸出

更新於:2021 年 8 月 4 日

4K+ 瀏覽

開啟你的職業

完成課程以獲得認證

開始
廣告