怎么得到混淆矩阵并把它可视化

// A code block
from skelearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
pred = [1,0,1,2,3,1,0,1,2,3]
true = [1,0,1,2,3,1,0,2,2,3]
classes = list(set(true))
confusion = confusion_matrix(true,pred)
indices = range(len(confusion))
plt.xticks(indices,classes)
plt.yticks(indices,classes)
plt.colorbar()
for first in range(len(confusion)):
	for second in range(len(first)):
		plt.txt(first,second,confusion[first][second],va='center',ha='center')
plt.show()

// An highlighted block
from skelearn.metrics imp