python 混淆矩阵输出如何调整字体大小
时间: 2024-02-04 21:08:53 浏览: 154
在Python中,可以使用matplotlib库来绘制混淆矩阵,并通过调整字体大小来改变混淆矩阵的输出。下面是一个示例代码:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建混淆矩阵
confusion_matrix = np.array([[10, 3], [5, 8]])
# 绘制混淆矩阵
fig, ax = plt.subplots()
im = ax.imshow(confusion_matrix, cmap='Blues')
# 设置颜色条
bar = ax.figure.colorbar(im, ax=ax)
# 设置刻度标签
ax.set_xticks(np.arange(confusion_matrix.shape[1]))
ax.set_yticks(np.arange(confusion_matrix.shape[0]))
# 设置刻度标签的字体大小
ax.set_xticklabels(['类别0', '类别1'], fontsize=12)
ax.set_yticklabels(['类别0', '类别1'], fontsize=12)
# 在每个单元格中显示数值
for i in range(confusion_matrix.shape[0]):
for j in range(confusion_matrix.shape[1]):
text = ax.text(j, i, confusion_matrix[i, j], ha='center', va='center', color='black', fontsize=12)
# 设置图像标题
ax.set_title('混淆矩阵', fontsize=14)
# 显示图像
plt.show()
```
在上述代码中,我们使用`imshow`函数绘制混淆矩阵,并使用`set_xticklabels`和`set_yticklabels`函数来设置刻度标签的字体大小。通过调整`fontsize`参数的值,可以改变字体的大小。最后,使用`text`函数在每个单元格中显示数值。
阅读全文