如何在 matplotlib 中绘制混淆矩阵?

matplotlibserver side programmingprogramming更新于 2025/5/13 0:07:17

使用 imshow 方法,我们可以创建一个输入数组维度为 (5, 5) 的图像。之后,我们可以使用 xticks 和 yticks 方法标记轴上的刻度。

步骤

  • 返回半开区间 [5, 5) 内的随机浮点数,插值="最近"。

  • 使用步骤 1 的数据将数据显示为图像,即在 2D 常规栅格上。

  • 使用 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()

输出


相关文章