MATLAB | 好看的相关系数矩阵图绘制
又是一个长长的工具类函数，创作编写不易，如果有用的话点赞,转发走起~
绘制超好看的相关系数矩阵图！工具函数由于过长将放在最后展示，以下先展示使用方式和绘制效果：
使用效果
使用教程
0 数据准备
数据就多列的数值矩阵即可：
X=[-2.8461 -0.9470 -0.1491 -1.6213 -1.1106 0.6428 2.7456 -1.0897
-0.7024 -0.1509 -1.9612 -0.9706 1.0950 0.3940 1.6108 0.6323
0.3761 -1.1815 -0.3485 -1.2662 -0.8673 -1.6869 -0.8052 1.4172
0.1717 0.0230 -0.1417 -0.0084 -1.2139 -0.6537 0.1804 0.5806
-1.5339 0.0020 -2.4430 -0.5684 0.0198 -0.2628 -0.0488 1.3821
-0.6906 0.6501 -0.2527 0.1571 0.7983 2.0210 -0.5023 -0.3868
0.7227 0.4469 1.1316 0.2165 1.9895 -0.9230 0.2779 0.1402
-1.1487 -0.1272 0.0974 0.7397 -1.1794 1.4483 0.5348 0.9840
2.4263 0.3561 -0.5850 -0.0829 0.9081 0.6718 3.1348 -1.6806
1.4553 -0.7559 1.3223 -0.1616 1.6948 -0.0941 0.2619 0.0846
1.1833 2.0956 -0.4242 -1.1503 0.4986 -0.7468 -1.1212 -1.4685
1.2845 -0.3788 -0.2780 3.2273 0.2187 -1.7527 -0.5731 0.8619
0.9713 0.3849 0.4713 1.3746 1.3742 -0.0587 0.5780 -0.0382
1.4192 0.4118 1.0904 0.6697 1.1387 -1.4548 -1.4595 -0.6207
-0.1758 -0.0541 3.3090 0.4501 1.5922 -2.0072 -0.5666 -1.0384
3.1798 1.2233 2.1071 0.3412 0.6792 -0.4604 0.3374 -0.6348
2.6781 0.9919 2.2081 2.3743 1.0871 0.2585 -0.6621 0.1306
1.6062 3.3028 2.5823 2.1885 2.4703 -1.1143 0.1006 1.9326
-0.0094 2.4318 3.1476 1.9018 2.9599 -0.2546 -1.3590 -2.0736
2.0687 1.5410 1.1177 2.2240 2.3615 -0.5323 -1.0246 -0.2873];
为了减少之后代码的篇幅，已将此矩阵存入XData.mat文件，后文使用如下方式导入数据：
load XData.mat
1 基本使用
对，就两行！
load XData.mat
CMP=corrMatPlot(X);
CMP=CMP.draw();
2 修改标签
理论上方法有三种，以下三种方式等价(推荐第二种)：
load XData.mat
CMP=corrMatPlot(X,'Labels',{'welcome','to','follow','my','wechat','official','account','slandarer'});
CMP=CMP.draw();
load XData.mat
CMP=corrMatPlot(X);
CMP=CMP.setLabelStr({'welcome','to','follow','my','wechat','official','account','slandarer'});
CMP=CMP.draw();
load XData.mat
CMP=corrMatPlot(X);
CMP=CMP.draw();
CMP.setLabelStr({'welcome','to','follow','my','wechat','official','account','slandarer'});
3 修改colormap
上面三种修改方式依旧适用，但为了节省篇幅只介绍最推荐的一种，同时此方法接受三种格式数据，使用setColorMap函数修改配色：
3.1 matlab自带colormap
matlab 自带的colormap均可用：
load XData.mat
CMP=corrMatPlot(X);
CMP=CMP.setColorMap(copper);
CMP=CMP.draw();
copper：
pink:
summer:
3.2 mx3大小RGB数组
load XData.mat
CMP=corrMatPlot(X);
cList=[114,146,184;255,255,255;132,158,119]./255;
CMP=CMP.setColorMap(cList);
CMP=CMP.draw();
3.3 内置colormap
本工具函数类内内置了6种colormap，可以通过：
- obj.setColorMap(n)
的格式使用：
load XData.mat
CMP=corrMatPlot(X);
CMP=CMP.setColorMap(2);
CMP=CMP.draw();
内置配色1：
内置配色2：
内置配色3：
内置配色4：
内置配色5：
内置配色6：
4 上下三角模式
设置类的Format属性即可：
full完整矩阵图triu上三角矩阵图tril下三角矩阵图
load XData.mat
CMP=corrMatPlot(X,'Format','tril');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
5 系数展示格式
通过调整Type属性调整格式：
sq: 方形(默认)ssq: 含文本方形pie: 饼图circ: 圆形oval: 椭圆形
load XData.mat
CMP=corrMatPlot(X,'Type','pie');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
全部格式展示(上三角)
sq:
ssq:
pie:
circ:
oval:
6 标签格式修改
使用：
- setXLabel
- setYLabel
函数分别修饰X轴Y轴标签,text对象具有的属性均可被修饰，以下展示修饰字体，字号及颜色：
load XData.mat
CMP=corrMatPlot(X,'Format','triu','Type','pie');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
CMP.setXLabel('Color',[.8,0,0],'FontName','Cambria','FontSize',15)
CMP.setYLabel('Color',[0,0,.8],'FontName','Cambria','FontSize',15)
7 框格式修改
使用setBox函数修饰框样式,Line对象具有的属性均可被修饰，以下展示加粗并更改颜色：
load XData.mat
CMP=corrMatPlot(X,'Format','triu','Type','pie');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
CMP.setBox('LineWidth',2,'Color',[.8,0,0])
7 内容物修饰
使用setPatch修改内容物格式，Patch对象具有的属性均可被修饰，以下展示加粗边缘，修改边缘颜色：
load XData.mat
CMP=corrMatPlot(X,'Format','triu','Type','pie');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
CMP.setPatch('LineWidth',2,'EdgeColor',[0,0,.8])
7 数值标签修饰
使用setCorrTxt函数进行修饰，仅ssq格式的图像可修饰：
load XData.mat
CMP=corrMatPlot(X,'Format','triu','Type','ssq');
CMP=CMP.setColorMap(1);
CMP=CMP.draw();
CMP.setCorrTxt('Color',[0,0,.8],'FontName','Cambria','FontSize',12)
工具函数完整代码
classdef corrMatPlot
% @author : slandarer
% gzh : slandarer随笔
% 使用示例：
% =========================================================================
% X=[-2.8461 -0.9470 -0.1491 -1.6213 -1.1106 0.6428 2.7456 -1.0897
% -0.7024 -0.1509 -1.9612 -0.9706 1.0950 0.3940 1.6108 0.6323
% 0.3761 -1.1815 -0.3485 -1.2662 -0.8673 -1.6869 -0.8052 1.4172
% 0.1717 0.0230 -0.1417 -0.0084 -1.2139 -0.6537 0.1804 0.5806
% -1.5339 0.0020 -2.4430 -0.5684 0.0198 -0.2628 -0.0488 1.3821
% -0.6906 0.6501 -0.2527 0.1571 0.7983 2.0210 -0.5023 -0.3868
% 0.7227 0.4469 1.1316 0.2165 1.9895 -0.9230 0.2779 0.1402
% -1.1487 -0.1272 0.0974 0.7397 -1.1794 1.4483 0.5348 0.9840
% 2.4263 0.3561 -0.5850 -0.0829 0.9081 0.6718 3.1348 -1.6806
% 1.4553 -0.7559 1.3223 -0.1616 1.6948 -0.0941 0.2619 0.0846
% 1.1833 2.0956 -0.4242 -1.1503 0.4986 -0.7468 -1.1212 -1.4685
% 1.2845 -0.3788 -0.2780 3.2273 0.2187 -1.7527 -0.5731 0.8619
% 0.9713 0.3849 0.4713 1.3746 1.3742 -0.0587 0.5780 -0.0382
% 1.4192 0.4118 1.0904 0.6697 1.1387 -1.4548 -1.4595 -0.6207
% -0.1758 -0.0541 3.3090 0.4501 1.5922 -2.0072 -0.5666 -1.0384
% 3.1798 1.2233 2.1071 0.3412 0.6792 -0.4604 0.3374 -0.6348
% 2.6781 0.9919 2.2081 2.3743 1.0871 0.2585 -0.6621 0.1306
% 1.6062 3.3028 2.5823 2.1885 2.4703 -1.1143 0.1006 1.9326
% -0.0094 2.4318 3.1476 1.9018 2.9599 -0.2546 -1.3590 -2.0736
% 2.0687 1.5410 1.1177 2.2240 2.3615 -0.5323 -1.0246 -0.2873];
%
% CMP=corrMatPlot(X,'Format','triu','Type','pie');
% CMP=CMP.setColorMap(1);
% CMP=CMP.draw();
% CMP.setLabelStr({'welcome','to','follow','my','wechat','official','account','slandarer'})
properties
ax;drawn=false;
arginList={'Labels','colorMap','Format','Type'}
Labels={};
colorMap={};
baseCM={[189, 53, 70;255,255,255; 97, 97, 97]./255,...
[113,161,195;255,255,255;228,103, 38]./255,...
[ 28,127,119;255,255,255;204,157, 80]./255,...
[130,130,255;255,255,255;255,133,133]./255,...
[209,58,78;253,203,121;254,254,189;198,230,156;63,150,181]./255,...
[243,166, 72;255,255,255;133,121,176]./255};
Format='full'% full:完整矩阵 triu:上三角 tril:下三角
Type='sq' % sq : 方形(默认)
% ssq : 含文本方形
% pie : 饼图
% circ : 圆形
% oval : 椭圆形
XData;num;corrMat;
% -----------------------------------------------------------------
boxHdl;txtHdl;
RLabelHdl;CLabelHdl;
colorbarHdl
matrixHdl
end
methods
function obj=corrMatPlot(varargin)
if isa(varargin{1},'matlab.graphics.axis.Axes')
obj.ax=varargin{1};varargin(1)=[];
else
obj.ax=gca;
end
% 获取版本信息
tver=version('-release');
verMatlab=str2double(tver(1:4))+(abs(tver(5))-abs('a'))/2;
if verMatlab<2017
hold on
else
hold(obj.ax,'on')
end
% -------------------------------------------------------------
obj.colorMap=obj.baseCM{1};
obj.XData=varargin{1};varargin(1)=[];
obj.corrMat=corr(obj.XData);
obj.num=size(obj.corrMat,1);
% 获取其他信息
for i=1:(length(varargin)-1)
tid=ismember(obj.arginList,varargin{i});
if any(tid)
obj.(obj.arginList{tid})=varargin{i+1};
end
end
if isempty(obj.Labels)
for i=1:obj.num
obj.Labels{i}=['class ',num2str(i)];
end
end
help corrMatPlot
end
% =========================================================================
function obj=draw(obj)
obj.ax.XLim=[.5,obj.num+.8];
obj.ax.YLim=[.5,obj.num+.5];
obj.ax.XTick=[];
obj.ax.YTick=[];
obj.ax.XColor='none';
obj.ax.YColor='none';
obj.ax.PlotBoxAspectRatio=[1,1,1];
% 绘制boxHdl
if strcmp(obj.Type(end-1:end),'sq');vb0='off';else,vb0='on';end
switch obj.Format
case 'full'
bX=[repmat([.5,obj.num+.5],[obj.num+1,1]),ones([obj.num+1,1]).*nan];
bY=[repmat((.5:1:obj.num+.5)',[1,2]),ones([obj.num+1,1]).*nan];
bXX=[bX;bY]';bYY=[bY;bX]';bXX=bXX(:);bYY=bYY(:);
obj.boxHdl=plot(bXX,bYY,'LineWidth',.8,'Color',[1,1,1].*.85,'Visible',vb0);
case 'tril'
bX=[.5.*ones([obj.num+1,1]),(0:obj.num)'+1.5,ones([obj.num+1,1]).*nan];
bX(end,2)=bX(end,2)-1;
bY=[repmat((obj.num+.5:-1:.5)',[1,2]),ones([obj.num+1,1]).*nan];
bXX=[bX;bY]';bYY=[bY;bX]';bXX=bXX(:);bYY=bYY(:);
obj.boxHdl=plot(bXX,bYY,'LineWidth',.8,'Color',[1,1,1].*.85,'Visible',vb0);
case 'triu'
bX=[(obj.num+.5).*ones([obj.num+1,1]),(obj.num:-1:0)'-.5,ones([obj.num+1,1]).*nan];
bX(end,2)=bX(end,2)+1;
bY=[repmat((.5:1:obj.num+.5)',[1,2]),ones([obj.num+1,1]).*nan];
bXX=[bX;bY]';bYY=[bY;bX]';bXX=bXX(:);bYY=bYY(:);
obj.boxHdl=plot(bXX,bYY,'LineWidth',.8,'Color',[1,1,1].*.85,'Visible',vb0);
end
colorFunc=colorFuncFactory(obj.colorMap);
colormap(colorFunc(linspace(-1,1,100)));caxis([-1,1])
cb=colorbar();
% 绘制colorbar
cb.Limits=[-1.02,1.02];
cb.TickDirection='out';
cb.LineWidth=.8;
cb.Ticks=-1:.2:1;
cb.FontName='Arial';
cb.FontSize=11;
obj.colorbarHdl=cb;
for row=1:obj.num
for col=1:obj.num
tValue=obj.corrMat(row,col);
tColor=colorFunc(tValue);
tGray=1-rgb2gray(tColor);
baseT=linspace(0,2*pi,500);
thetaMat=[1,-1;1,1].*sqrt(2)./2;
vb1='on';
if strcmp(obj.Format,'triu')&&row>col,vb1='off';end
if strcmp(obj.Format,'tril')&&row<col,vb1='off';end
switch obj.Type
case 'sq' % 方形
baseSqX=[-.5,.5,.5,-.5].*0.95;
baseSqY=[-.5,-.5,.5,.5].*0.95;
obj.matrixHdl(row,col).f=fill(baseSqX+col,baseSqY+1+obj.num-row,...
tColor,'EdgeColor','none','Visible',vb1);
case 'ssq' % 含文本方形
baseSqX=[-.5,.5,.5,-.5].*0.95;
baseSqY=[-.5,-.5,.5,.5].*0.95;
obj.matrixHdl(row,col).f=fill(baseSqX+col,baseSqY+1+obj.num-row,...
tColor,'EdgeColor','none','Visible',vb1);
obj.matrixHdl(row,col).t=text(col,1+obj.num-row,...
sprintf('%.2f',obj.corrMat(row,col)),'FontName','Arial','FontSize',10,...
'HorizontalAlignment','center','Color',tGray,'Visible',vb1);
case 'pie' % 饼图
baseCircX=cos(baseT).*.92.*.5;
baseCircY=sin(baseT).*.92.*.5;
obj.matrixHdl(row,col).c=fill(baseCircX+col,baseCircY+1+obj.num-row,...
[1,1,1],'EdgeColor',[1,1,1].*.3,'LineWidth',.8,'Visible',vb1);
baseTheta=linspace(pi/2,pi/2+tValue.*2.*pi,200);
basePieX=[0,cos(baseTheta).*.92.*.5];
basePieY=[0,sin(baseTheta).*.92.*.5];
obj.matrixHdl(row,col).f=fill(basePieX+col,basePieY+1+obj.num-row,...
tColor,'EdgeColor',[1,1,1].*.3,'lineWidth',.8,'Visible',vb1);
case 'circ' % 圆形
baseR=(.25+.6.*abs(tValue)).*.5;
baseCircX=cos(baseT).*baseR;
baseCircY=sin(baseT).*baseR;
obj.matrixHdl(row,col).f=fill(baseCircX+col,baseCircY+1+obj.num-row,...
tColor,'EdgeColor','none','Visible',vb1);
case 'oval' % 椭圆形
baseA=1+(tValue<=0).*tValue;
baseB=1-(tValue>=0).*tValue;
baseOvalX=cos(baseT).*.98.*.5.*baseA;
baseOvalY=sin(baseT).*.98.*.5.*baseB;
baseOvalXY=thetaMat*[baseOvalX;baseOvalY];
obj.matrixHdl(row,col).f=fill(baseOvalXY(1,:)+col,baseOvalXY(2,:)+1+obj.num-row,...
tColor,'EdgeColor',[1,1,1].*.3,'lineWidth',.8,'Visible',vb1);
end
end
end
tfig=obj.ax.Parent;
tfig.Position(2)=tfig.Position(2)./2;
tfig.Position(4)=tfig.Position(3);
tfig.Color=[1,1,1];
for row=1:obj.num
obj.RLabelHdl(row)=text(.42+(row-1).*strcmp(obj.Format,'triu'),obj.num+1-row,...
obj.Labels{row},'HorizontalAlignment','right',...
'FontName','Arial','FontSize',12);
end
for col=1:obj.num
obj.CLabelHdl(col)=text(col-.1,obj.num+.75-(col-1).*strcmp(obj.Format,'tril'),...
obj.Labels{col},'HorizontalAlignment','left',...
'FontName','Arial','FontSize',12,'Rotation',30);
end
obj.drawn=true;
% -------------------------------------------------------------
% 渐变色句柄生成函数
function colorFunc=colorFuncFactory(colorList)
x=linspace(-1,1,size(colorList,1));
y1=colorList(:,1);y2=colorList(:,2);y3=colorList(:,3);
colorFunc=@(X)[interp1(x,y1,X,'linear')',...
interp1(x,y2,X,'linear')',interp1(x,y3,X,'linear')'];
end
end
% =========================================================================
% 预设/设置colormap
function obj=setColorMap(obj,CM)
if length(CM)==1
obj.colorMap=obj.baseCM{CM};
else
obj.colorMap=CM;
end
if obj.drawn
colorFunc=colorFuncFactory(obj.colorMap);
colormap(colorFunc(linspace(-1,1,100)));
for row=1:obj.num
for col=1:obj.num
tValue=obj.corrMat(row,col);
tColor=colorFunc(tValue);
tGray=1-rgb2gray(tColor);
set(obj.matrixHdl(row,col).f,'FaceColor',tColor);
if strcmp(obj.Type,'ssq')
set(obj.matrixHdl(row,col).t,'Color',tGray)
end
end
end
end
% -------------------------------------------------------------
% 渐变色句柄生成函数
function colorFunc=colorFuncFactory(colorList)
x=linspace(-1,1,size(colorList,1));
y1=colorList(:,1);y2=colorList(:,2);y3=colorList(:,3);
colorFunc=@(X)[interp1(x,y1,X,'linear')',...
interp1(x,y2,X,'linear')',interp1(x,y3,X,'linear')'];
end
end
% XY轴标签设置
function obj=setLabelStr(obj,LB)
obj.Labels=LB;
if obj.drawn
for i=1:obj.num
set(obj.RLabelHdl(i),'String',obj.Labels{i});
set(obj.CLabelHdl(i),'String',obj.Labels{i});
end
end
end
function obj=setXLabel(obj,varargin)
for i=1:obj.num
set(obj.CLabelHdl(i),varargin{:})
end
end
function obj=setYLabel(obj,varargin)
for i=1:obj.num
set(obj.RLabelHdl(i),varargin{:})
end
end
% patch对象设置
function obj=setPatch(obj,varargin)
for row=1:obj.num
for col=1:obj.num
set(obj.matrixHdl(row,col).f,varargin{:});
if strcmp(obj.Type,'pie')
set(obj.matrixHdl(row,col).c,varargin{:},'FaceColor',[1,1,1])
end
end
end
end
% 设置数值文本
function obj=setCorrTxt(obj,varargin)
if strcmp(obj.Type,'ssq')
for row=1:obj.num
for col=1:obj.num
set(obj.matrixHdl(row,col).t,varargin{:})
end
end
end
end
% 设置框样式
function obj=setBox(obj,varargin)
set(obj.boxHdl,varargin{:},'Visible','on')
end
end
% @author : slandarer
% gzh : slandarer随笔
end
另：试了一下更多组数据绘图，效果还行：
完
未经允许本代码请勿作商业用途，引用的话可以引用我file exchange上的链接，可使用如下格式：
Zhaoxu Liu (2022). corr plot 相关系数矩阵图 (https://www.mathworks.com/matlabcentral/fileexchange/117100-corr-plot), MATLAB Central File Exchange. 检索来源 2022/9/3.
若转载请保留以上file exchange链接及本文链接！！！
以上已经为全部代码，但懒得复制的话:
【相关系数矩阵图】
更新时会跟进更新以下连接：
【链接】：https://pan.baidu.com/s/1URle1b3dwdsE-9gtj5Ru-g?pwd=slan
【提取码】：slan