python中画混淆矩阵

在Python中，可以使用matplotlib和sklearn库来绘制混淆矩阵。其中，sklearn.metrics.ConfusionMatrixDisplay类可以方便地绘制混淆矩阵。下面是一个示例代码：

from sklearn.metrics import confusion_matrix
from sklearn.metrics import ConfusionMatrixDisplay
import matplotlib.pyplot as plt

y_true = [2, 0, 2, 2, 0, 1]
y_pred = [0, 0, 2, 2, 0, 2]

cm = confusion_matrix(y_true, y_pred)
disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=["ant", "bird", "cat"])
disp.plot(cmap=plt.cm.Blues)
plt.title("Confusion Matrix")
plt.show()

其中，y_true和y_pred分别表示真实标签和预测标签，cm表示混淆矩阵，disp表示ConfusionMatrixDisplay对象，display_labels表示标签名称，cmap表示颜色映射。