如何在 matplotlib 中繪製混淆矩陣？

透過 imshow 方法，我們可以使用輸入 (5, 5) 陣列維度建立影像。之後，我們可以使用 xticks 和 yticks 方法來標記軸上的刻度。

步驟

  • 以半開區間 [5, 5) 和插值='nearest' 返回隨機浮點數。

  • 將資料顯示為影像，也就是說，在一個 2D 正則光柵上，以步驟 1 方式顯示資料。

  • 使用 xticks 方法獲取或設定 x 軸的當前刻度位置和標籤。

  • 使用 yticks 方法獲取或設定 y 軸的當前刻度位置和標籤。

  • 使用 plt.show() 來顯示圖形。

示例

import matplotlib.pyplot as plt
import numpy as np

plt.imshow(np.random.random((5, 5)), interpolation='nearest')
plt.xticks(np.arange(0, 5), ['A', 'B', 'C', 'D', 'E'])
plt.yticks(np.arange(0, 5), ['1', '2', '3', '4', '5'])

plt.show()

輸出

更新於： 15-3-2021

824 次瀏覽

開啟您的職業生涯

完成課程並取得認證

開始
廣告