matplotlb 混淆矩阵论文用 python 画matlab产生的混淆矩阵

TinaO-O

于 2019-03-05 13:59:57 发布

阅读量1.8k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python tex 使用latex撰写论文

本文链接：https://blog.youkuaiyun.com/u013249853/article/details/88179684

本文介绍如何使用Python的matplotlib库来绘制MATLAB生成的混淆矩阵，包括设置数值四舍五入、调整坐标轴以及设定显示阈值，如仅显示大于15的值。示例中，数据来源于MATLAB的.mat文件，转换为Python的字典和numpy数组，最终保存为PDF以供论文使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

论文当中经常会见到：

这样的混淆矩阵，之前我们讨论过怎么将其放到latex中，这里讨论下，怎么设置四舍五入，怎么更改坐标，怎么限制显示阈值。比如上面这张图，小于15的就不显示。我们的数据来源是一个matlab的.mat文件，读入后会自动生成字典，并且该混淆矩阵是一个np.arry。

直接看代码注释：

from numpy import *
import matplotlib.pyplot as plt
from pylab import *
import scipy.io
import math
#这里读入了matlab生成的.mat数据
data = scipy.io.loadmat('D:/GMM/nyuv1/con_82.mat')

#conf_arr = [[33,2,0,0,0,0,0,0,0,1,3], [3,31,0,0,0,0,0,0,0,0,0], [0,4,41,0,0,0,0,0,0,0,1], [0,1,0,30,0,6,0,0,0,0,1], [0,0,0,0,38,10,0,0,0,0,0], [0,0,0,3,1,39,0,0,0,0,4], [0,2,2,0,4,1,31,0,0,0,2], [0,1,0,0,0,0,0,36,0,2,0], [0,0,0,0,0,0,1,5,37,5,1], [3,0,0,0,0,0,0,0,0,39,0], [0,0,0,0,0,0,0,0,0,0,38] ]
#以字典的形式读入其中存的一个变量con_re_mm，该变量就是个数组，混淆矩阵，13*13的
data = data['con_re_mm']
conf_arr = data;
#这里四舍五入的取整，直接改变了数据形式，不然画出来的图全是99.0 88.0而不是99 88
conf_arr = conf_arr.asty