python/matplotlib绘制混淆矩阵图


from sklearn.metrics import confusion_matrix, f1_score, multilabel_confusion_matrix
from operator import truediv
import numpy 
import matplotlib.pyplot as plt 

if __name__ == "__main__":

    confusion = numpy.array((
        [0, 2,  3 , 5, 3, 6],
        [1, 0,  1 ,  0,  0 ,  1],
        [  6 ,  1,  0 ,  4 ,  1 ,  2],
        [  7 ,  1 ,  0 ,0 , 10 ,  4],
        [  2 ,  2 ,  1 , 16 ,0 ,  9],
        [  8 ,  3,   4 ,  9 ,  6 ,0]))
    # plt.matshow(cf_matrix,cmap=plt.cm.gray)
    # plt.savefig('true.jpg')
    # row_sum = numpy.sum(cf_matrix,axis=1)
    # err_matrix = cf_matrix / row_sum
    # numpy.fill_diagonal(err_matrix,0)

    # # plt.matshow(err_matrix,cmap=plt.cm.gray)
    # plt.matshow(err_matrix)
    # plt.savefig('false.jpg')
    # 热度图,后面是指定的颜色块,可设置其他的不同颜色
    plt.imshow(confusion, cmap=plt.cm.Blues)
    # plt.imshow(confusion)
    # ticks 坐标轴的坐标点
    # label 坐标轴标签说明
    indices = range(len(confusion))
    
 
    # 第一个是迭代对象,表示坐标的显示顺序,第二个参数是坐标轴显示列表
    plt.xticks(indices, [0, 1, 2, 3, 4, 5])
    plt.yticks(indices, [0, 1, 2, 3, 4, 5])

    plt.colorbar()

    plt.xlabel('Predict')
    plt.ylabel('GT')
    plt.title('cf_matrix')

   # plt.rcParams两行是用于解决标签不能显示汉字的问题
    plt.rcParams['font.sans-serif']=['SimHei']
    plt.rcParams["font.family"] = "sans-serif"
    plt.rcParams['axes.unicode_minus'] = False


    # 显示数据
    for first_index in range(len(confusion)):    #第几行
        for second_index in range(len(confusion[first_index])):    #第几列
            # plt.text(first_index, second_index, confusion[first_index][second_index])
            plt.text(first_index, second_index, confusion[second_index][first_index])

    # 在matlab里面可以对矩阵直接imagesc(confusion)
    # 显示
    plt.show()
    plt.savefig('false2.jpg')

    原文作者:邦戈栗子
    原文地址: https://blog.csdn.net/qq_42189368/article/details/123424089
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞