如何在 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()
輸出
廣告