Python 用有意义的行和列标签绘制混淆矩阵

关于我的混乱矩阵布局的问题。我想将变量标签对应的数字“0”、“1”和“2”替换为实际变量标签，即“裸土”、“玉米”和“其他作物”。 谢谢你的帮助

CM=混淆矩阵（y_测试，y_预测）
CM=sns.heatmap（CM，square=True，cmap=“coolwarm”，线宽=0.5，annot=True）
打印标题（“型号混淆矩阵”，字体大小=20）
plt.xlabel（'Predict'，fontsize=15）
plt.ylabel（'Real'，fontsize=15）
plt.show（）

---

试试

plt.yticks

，

plt.xticks

：

import matplotlib.pyplot as plt
sns.heatmap(confusion_matrix(y_test,y_pred),annot=True)
plt.yticks(np.arange(3)+0.5,('X1','X2','X3'), fontsize="10")
plt.xticks(np.arange(3)+0.5,('X1','X2','X3'), fontsize="10" )

您可以使用，也可以使用。