Python绘制混淆矩阵,汉字显示label
程序员文章站
2022-03-20 16:34:16
1. 在计算出混淆矩阵之后,想自己绘制图形(并且在图形上显示汉字),可用 2. 除了设置plt.rcParams的特征之外,显示汉字的第二个方式 —— 设置fm.FontProperties属性 设置前: 设置后: ......
1. 在计算出混淆矩阵之后,想自己绘制图形(并且在图形上显示汉字),可用
#coding=utf-8 import matplotlib.pyplot as plt import numpy as np confusion = np.array(([91,0,0],[0,92,1],[0,0,95])) # 热度图,后面是指定的颜色块,可设置其他的不同颜色 plt.imshow(confusion, cmap=plt.cm.blues) # ticks 坐标轴的坐标点 # label 坐标轴标签说明 indices = range(len(confusion)) # 第一个是迭代对象,表示坐标的显示顺序,第二个参数是坐标轴显示列表 #plt.xticks(indices, [0, 1, 2]) #plt.yticks(indices, [0, 1, 2]) plt.xticks(indices, ['圆形', '三角形', '方形']) plt.yticks(indices, ['圆形', '三角形', '方形']) plt.colorbar() plt.xlabel('预测值') plt.ylabel('真实值') plt.title('混淆矩阵') # plt.rcparams两行是用于解决标签不能显示汉字的问题 plt.rcparams['font.sans-serif']=['simhei'] 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]) # 在matlab里面可以对矩阵直接imagesc(confusion) # 显示 plt.show()
2. 除了设置plt.rcparams的特征之外,显示汉字的第二个方式 —— 设置fm.fontproperties属性
import matplotlib.font_manager as fm import matplotlib.pyplot as plt import numpy as np myfont = fm.fontproperties(fname='c:/windows/fonts/simsun.ttc') confusion = np.array(([91,0,0],[0,92,1],[0,0,95])) plt.plot(confusion[0]) plt.xlabel(u'索引',fontproperties=myfont) plt.ylabel(u'值',fontproperties=myfont) plt.show()
设置前:
设置后:
下一篇: Java的含义