如何在 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 次浏览

开启您的职业生涯

完成课程并取得认证

开始
广告