用Python生成一个混淆矩阵图,但是要求不适用热力图,就是数字是黑色的,网格是黑色的,字体是黑色的,其他地方是透明的
时间: 2023-06-18 16:07:45 浏览: 100
可以使用Matplotlib库来绘制混淆矩阵图,并设置字体和颜色。以下是一个示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 混淆矩阵,这里是一个示例
confusion_matrix = np.array([[100, 20, 30],
[10, 200, 20],
[30, 20, 150]])
# 设置图像大小和DPI
fig = plt.figure(figsize=(5, 5), dpi=100)
# 绘制混淆矩阵图
plt.imshow(confusion_matrix, cmap='gray')
# 设置网格线颜色
plt.grid(color='black', linestyle='-', linewidth=1)
# 设置坐标轴标签和刻度
plt.xticks([0, 1, 2], ['Class 1', 'Class 2', 'Class 3'])
plt.yticks([0, 1, 2], ['Class 1', 'Class 2', 'Class 3'])
plt.xlabel('Predicted Label')
plt.ylabel('True Label')
# 设置每个网格的数字标签
for i in range(3):
for j in range(3):
plt.text(j, i, str(confusion_matrix[i, j]), ha='center', va='center', color='black')
# 去掉图像周围的边框
plt.gca().spines['top'].set_visible(False)
plt.gca().spines['right'].set_visible(False)
plt.gca().spines['bottom'].set_visible(False)
plt.gca().spines['left'].set_visible(False)
# 显示图像
plt.show()
```
运行以上代码,就可以得到一个黑色的混淆矩阵图,网格和数字都是黑色的,其他地方是透明的。可以根据需要调整图像大小和字体大小等参数。
阅读全文