python如何绘制混淆矩阵
时间: 2023-08-14 10:11:13 浏览: 54
要使用Python绘制混淆矩阵,你可以使用一些常用的数据可视化库,如matplotlib和seaborn。以下是一个简单的示例代码,展示了如何使用这两个库绘制混淆矩阵:
```python
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
# 创建混淆矩阵
confusion_matrix = np.array([[50, 10, 0],
[5, 30, 5],
[10, 5, 40]])
# 创建标签
labels = ['Class 0', 'Class 1', 'Class 2']
# 绘制混淆矩阵
plt.figure(figsize=(8, 6))
sns.heatmap(confusion_matrix, annot=True, fmt='d', cmap='Blues', xticklabels=labels, yticklabels=labels)
plt.title('Confusion Matrix')
plt.xlabel('Predicted Labels')
plt.ylabel('True Labels')
plt.show()
```
这段代码首先创建了一个3x3的混淆矩阵,然后使用`sns.heatmap()`函数将其可视化为一个热图。`annot=True`表示在热图中显示数值,`fmt='d'`表示使用整数格式显示数值。`cmap='Blues'`设置颜色映射为蓝色。通过设置`xticklabels`和`yticklabels`,可以指定横轴和纵轴的标签。
运行代码后,你将得到一个绘制好的混淆矩阵图。你可以根据自己的需求进行调整和修改,例如,修改矩阵的大小、颜色映射等。