利用python绘制混淆矩阵，直接附代码。
from sklearn.metrics import confusion_matrix
import pandas as pd
import matplotlib.pyplot as plt
y_pred = [] # ['2','2','3','1','4'] # 类似的格式
y_true = [] # ['0','1','2','3','4'] # 类似的格式
# 对上面进行赋值
C=confusion_matrix(y_true, y_pred, labels=['0','1','2','3','4']) # 可将'1'等替换成自己的类别，如'cat'。
plt.matshow(C, cmap=plt.cm.Reds) # Blues Reds Purples Greens
# plt.colorbar()
for i in range(len(C)):
for j in range(len(C)):
plt.annotate(C[j,i], xy=(i, j), horizontalalignment='center', verticalalignment='center')
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()
效果图：
如果需要切换不同的颜色，如：
参考：
python colorbar 颜色大全
