绘制效果
请叫我平平无奇的绘图天才,这期一次性带来俩工具函数,分别是两层和三层的环形树状图,可以用来展示数据分类和层级:
理顺层级关系属实理得头秃,代码编写不易点个赞
叭!!
由于工具函数过长,将被放在最后展示,以下将先展示函数用法。
双层环形树状图
基本使用
处理的数据对象是两列的元胞数组,数组内容物为字符串,像下面这样:
treeList={'竞赛','数学竞赛';'竞赛','英语竞赛';'竞赛','美赛';'竞赛','数模国赛';'竞赛','网络挑战赛';'竞赛','英语翻译';
'编程','python';'编程','MATLAB';'编程','C#';'编程','C++';'编程','C';'编程','java';'编程','js';'编程','R';'编程','html';
'编程','php';'编程','rust';'编程','julia';'编程','perl';'编程','carbon';'编程','lisp';
'技能','office';'技能','LaTeX';'技能','PS';'技能','PR';'技能','excel';'技能','视频剪辑';'技能','bat';
'基础知识','代数集合';'基础知识','傅里叶';'基础知识','抽代';'基础知识','数分';'基础知识','高代';'基础知识','解几';
'基础知识','实变';'基础知识','复变';'基础知识','运筹学';'基础知识','泛函';'基础知识','张量分解';'基础知识','概率论';
'开发工具','eclipse';'开发工具','git';'开发工具','svn';'开发工具','gitee';'开发工具','jupyter';'开发工具','macos';
'开发工具','postman';'开发工具','vscode';'开发工具','mdnice';'开发工具','pycharm';'开发工具','vim';'开发工具','svn';
'爱好','国画';'爱好','乒乓球';'爱好','折纸';'爱好','雕刻';'爱好','篮球';'爱好','足球'
'爱好','唱';'爱好','跳';'爱好','rap';'爱好','背带裤制作';'爱好','理发'};
CT=circleTree2(treeList);
CT=CT.draw();
好,画完结束了,当然线更多会比较好看,这里自己构造一些ABCD类再画个图:
treeList={[],[]};
k=1;
classNameSet={'A','B','C','D','E','F'};
classNumSet=[20,20,20,25,30,35];
for i=1:6
for j=1:classNumSet(i)
treeList{k,1}=['CLASS ',classNameSet{i}];
treeList{k,2}=[classNameSet{i},num2str(j)];
k=k+1;
end
end
CT=circleTree2(treeList);
CT=CT.draw();
修改配色
使用setColorN
设置各个类的颜色,比如将第2个类设为红色:
CT.setColorN(2,[.8,0,0]);
循环全部设置为灰色:
for i=1:6
CT.setColorN(i,[.8,.8,.8]);
end
修改字体
使用setLable1
和setLable1
修改两层文本字体,例如:
CT.setLable1('FontSize',16,'Color',[0,0,.8]);
CT.setLable2('FontSize',13,'Color',[0,.6,.8]);
三层环形树状图
基本使用
与上面双层环形树状图绘制方式一模一样,三层的不太好构造,这里直接从excel读取数据绘制,数据为往年诺贝尔奖获得者及其奖项、国籍
读取数据并绘制三层环形树状图:
treeList=readcell('data.xlsx');
treeList=treeList(:,1:3);
CT=circleTree3(treeList);
CT=CT.draw();
其他属性设置方式与之前的双层绘制函数完全相同,由于怕文字互相遮挡,故只展示占比大于5%的类,想要隐藏或显示全部类,可以设置文本的Visible
属性为on/off
:
CT.setLable2('Visible','on');
虽然有点乱,但是可以缩放查看细节:
完整代码
大量代码,前方高能!!
双层
classdef circleTree2
% @author : slandarer
% gzh : slandarer随笔
properties
ax,treeList
ClassLine,ChildLine
ClassScatter,MeanScatter
LabelHdl0,LabelHdl1
end
methods
function obj=circleTree2(treeList)
obj.ax=gca;hold on;
obj.treeList=treeList;
end
function obj=draw(obj)
% 坐标区域属性设置 ==============================================
obj.ax=gca;hold on
obj.ax.XLim=[-1,1];
obj.ax.YLim=[-1,1];
obj.ax.XTick=[];
obj.ax.YTick=[];
obj.ax.XColor='none';
obj.ax.YColor='none';
obj.ax.PlotBoxAspectRatio=[1,1,1];
fig=obj.ax.Parent;
if max(fig.Position(3:4))<600
fig.Position(3:4)=1.8.*fig.Position(3:4);
fig.Position(1:2)=fig.Position(1:2)./3;
end
% ax.LooseInset=[0,0,0,0];
% 理顺层级 =====================================================
classNameList=unique(obj.treeList(:,1));
childrenList{length(classNameList)}=[];
classSize(length(classNameList))=0;
for i=1:length(classNameList)
childrenList{i}=find(strcmp(classNameList{i},obj.treeList(:,1)));
classSize(i)=length(childrenList{i});
end
% 开始绘图 =====================================================
sepTheta=2/30/length(classSize);
cumTheta=[0,28/30*cumsum(classSize)./sum(classSize)];
colorList=[127,91,93;187,128,110;197,173,143;59,71,111;104,95,126;76,103,86;112,112,124;
72,39,24;197,119,106;160,126,88;238,208,146]./255;
colorList=[colorList;rand(length(classSize),3)];
colorList=colorList(4:end,:);
for i=1:length(classSize)
thetaList=linspace(sepTheta*i+cumTheta(i),sepTheta*i+cumTheta(i+1),classSize(i)).*2.*pi;
bX1=[];bY1=[];
for j=1:classSize(i)
oX1=[cos(thetaList(j)),cos(thetaList(j)).*0.9,cos(mean(thetaList)).*0.7,cos(mean(thetaList)).*0.45];
oY1=[sin(thetaList(j)),sin(thetaList(j)).*0.9,sin(mean(thetaList)).*0.7,sin(mean(thetaList)).*0.45];
bXY1=bezierCurve([oX1',oY1'],200);
bX1=[bX1;bXY1(:,1);nan];
bY1=[bY1;bXY1(:,2);nan];
nameList1=obj.treeList(:,2);
nameList1=nameList1(childrenList{i});
rotation=thetaList(j)/pi*180;
if rotation>90&&rotation<270
rotation=rotation+180;
obj.LabelHdl1(i,j)=text(cos(thetaList(j)).*1.03,sin(thetaList(j)).*1.03,nameList1{j},'Rotation',rotation,'HorizontalAlignment','right');
else
obj.LabelHdl1(i,j)=text(cos(thetaList(j)).*1.03,sin(thetaList(j)).*1.03,nameList1{j},'Rotation',rotation);
end
end
meanTheta=mean(thetaList);
obj.ChildLine(i)=plot(bX1,bY1,'Color',[colorList(i,:),.4],'LineWidth',1.5);
oX0=[cos(meanTheta).*0.46,cos(meanTheta).*0.44,0.05,0];
oY0=[sin(meanTheta).*0.46,sin(meanTheta).*0.44,-0.15,0];
bXY0=bezierCurve([oX0',oY0'],200);
obj.ClassLine(i)=plot(bXY0(:,1),bXY0(:,2),'Color',[colorList(i,:),.8],'LineWidth',1.5);
obj.ClassScatter(i)=scatter(cos(meanTheta).*0.45,sin(meanTheta).*0.45,'filled',...
'CData',colorList(i,:),'MarkerEdgeColor',[.4,.4,.4],'SizeData',60);
rotation=meanTheta/pi*180;
if rotation>90&&rotation<270
rotation=rotation+180;
obj.LabelHdl0(i)=text(cos(meanTheta).*0.48,sin(meanTheta).*0.48,classNameList{i},...
'Rotation',rotation,'HorizontalAlignment','right','FontSize',12,'VerticalAlignment','cap');
else
obj.LabelHdl0(i)=text(cos(meanTheta).*0.48,sin(meanTheta).*0.48,classNameList{i},...
'Rotation',rotation,'FontSize',12,'VerticalAlignment','cap');
end
end
obj.MeanScatter=scatter(0,0,'filled','CData',mean(colorList(1:length(classSize),:)),'MarkerEdgeColor',[.4,.4,.4],'SizeData',60);
disp(char([64 97 117 116 104 111 114 32 58 32 115 108 97 110 100 97 114 101 114]))
% 贝塞尔函数 ===================================================
function pnts=bezierCurve(pnts,N)
t=linspace(0,1,N);
p=size(pnts,1)-1;
coe1=factorial(p)./factorial(0:p)./factorial(p:-1:0);
coe2=((t).^((0:p)')).*((1-t).^((p:-1:0)'));
pnts=(pnts'*(coe1'.*coe2))';
end
end
% 修改颜色函数
function obj=setColorN(obj,n,color)
set(obj.ClassLine(n),'Color',[color,.8]);
set(obj.ChildLine(n),'Color',[color,.4]);
set(obj.ClassScatter(n),'MarkerFaceColor',color);
set(obj.ClassScatter(n),'CData',color);
tColorList=zeros(length(obj.ClassScatter),3);
for i=1:length(obj.ClassScatter)
tColorList(i,:)=get(obj.ClassScatter(i),'CData');
end
set(obj.MeanScatter,'MarkerFaceColor',mean(tColorList));
set(obj.MeanScatter,'CData',mean(tColorList));
end
% 修改第一层文字属性
function obj=setLable1(obj,varargin)
for i=1:length(obj.LabelHdl0)
set(obj.LabelHdl0(i),varargin{:});
end
end
% 修改第二层文字属性
function obj=setLable2(obj,varargin)
for i=1:size(obj.LabelHdl1,1)
for j=1:size(obj.LabelHdl1,2)
if obj.LabelHdl1(i,j)~=0
set(obj.LabelHdl1(i,j),varargin{:});
end
end
end
end
end
% Zhaoxu Liu (2022). Circular dendrogram 环形树状图
% (https://www.mathworks.com/matlabcentral/fileexchange/118325-circular-dendrogram),
% MATLAB Central File Exchange. 检索来源 2022/9/29.
end
三层
classdef circleTree3
% @author : slandarer
% gzh : slandarer随笔
properties
ax,treeList
ClassLine,ChildLine1,ChildLine2
MeanScatter,ClassScatter1,ClassScatter2
LabelHdl0,LabelHdl1,LabelHdl2
end
methods
function obj = circleTree3(treeList)
obj.ax=gca;hold on;
obj.treeList=treeList;
end
function obj=draw(obj)
% 坐标区域属性设置 ==============================================
obj.ax=gca;hold on
obj.ax.XLim=[-1,1];
obj.ax.YLim=[-1,1];
obj.ax.XTick=[];
obj.ax.YTick=[];
obj.ax.XColor='none';
obj.ax.YColor='none';
obj.ax.PlotBoxAspectRatio=[1,1,1];
fig=obj.ax.Parent;
if max(fig.Position(3:4))<600
fig.Position(3:4)=1.8.*fig.Position(3:4);
fig.Position(1:2)=fig.Position(1:2)./3;
end
% ax.LooseInset=[0,0,0,0];
% 理顺层级 =====================================================
classNameList=unique(obj.treeList(:,1));
childNameList1{length(classNameList)}={[]};
childrenList2{length(classNameList)}={[]};
for i=1:length(classNameList)
tList=obj.treeList(:,2);
tList=tList(strcmp(classNameList{i},obj.treeList(:,1)));
childNameList1{i}=unique(tList);
end
for i=1:length(classNameList)
for j=1:length(childNameList1{i})
tchildNameList=childNameList1{i};
tbool1=strcmp(classNameList{i},obj.treeList(:,1));
tbool2=strcmp(tchildNameList{j},obj.treeList(:,2));
childrenList2{i,j}=find(tbool1&tbool2);
classSize(i,j)=length(childrenList2{i,j});
end
end
cumSize=classSize';
colSize=cumSize(cumSize~=0)';
cumSize=cumsum(colSize);
% 开始绘图 =====================================================
sepTheta=2/30/length(cumSize);
cumTheta=[0,28/30.*cumSize./cumSize(end)];
colorList=[127,91,93;187,128,110;197,173,143;59,71,111;104,95,126;76,103,86;112,112,124;
72,39,24;197,119,106;160,126,88;238,208,146]./255;
colorList=[colorList;rand(length(classSize),3)];
colorList=colorList(4:end,:);
n=1;
for i=1:length(classNameList)
meanThetaList=[];
for j=1:length(childNameList1{i})
thetaList=linspace(sepTheta*n+cumTheta(n),sepTheta*n+cumTheta(n+1),classSize(i,j)).*2.*pi;
bX2=[];bY2=[];
for k=1:classSize(i,j)
oX2=[cos(thetaList(k)),cos(thetaList(k)).*0.95,cos(mean(thetaList)).*0.9,cos(mean(thetaList)).*0.7];
oY2=[sin(thetaList(k)),sin(thetaList(k)).*0.95,sin(mean(thetaList)).*0.9,sin(mean(thetaList)).*0.7];
bXY2=bezierCurve([oX2',oY2'],200);
bX2=[bX2;bXY2(:,1);nan];
bY2=[bY2;bXY2(:,2);nan];
end
obj.ChildLine2(i,j)=plot(bX2,bY2,'Color',[colorList(i,:),.4],'LineWidth',1.1);
meanThetaList=[meanThetaList,mean(thetaList)];
tchildNameList=childNameList1{i};
rotation=mean(thetaList)/pi*180;
if rotation>90&&rotation<270
rotation=rotation+180;
obj.LabelHdl1(n)=text(cos(mean(thetaList)).*0.7,sin(mean(thetaList)).*0.7,tchildNameList{j},...
'Rotation',rotation,'HorizontalAlignment','right','FontSize',11,'VerticalAlignment','bottom');
else
obj.LabelHdl1(n)=text(cos(mean(thetaList)).*0.7,sin(mean(thetaList)).*0.7,tchildNameList{j},...
'Rotation',rotation,'FontSize',11,'VerticalAlignment','cap');
end
if classSize(i,j)<sum(classSize(i,:))/20
set(obj.LabelHdl1(n),'Visible','off');
end
n=n+1;
end
obj.ClassScatter2(i)=scatter(cos(meanThetaList).*0.7,sin(meanThetaList).*0.7,'filled',...
'CData',colorList(i,:),'MarkerEdgeColor',[.4,.4,.4],'SizeData',20);
bX1=[];bY1=[];
for m=1:length(meanThetaList)
oX1=[cos(meanThetaList(m)).*0.7,cos(meanThetaList(m)).*0.65,cos(mean(meanThetaList)).*0.5,cos(mean(meanThetaList)).*0.35];
oY1=[sin(meanThetaList(m)).*0.7,sin(meanThetaList(m)).*0.65,sin(mean(meanThetaList)).*0.5,sin(mean(meanThetaList)).*0.35];
bXY1=bezierCurve([oX1',oY1'],200);
bX1=[bX1;bXY1(:,1);nan];
bY1=[bY1;bXY1(:,2);nan];
end
obj.ChildLine1(i)=plot(bX1,bY1,'Color',[colorList(i,:),.5],'LineWidth',1.2);
meanTheta=mean(meanThetaList);
oX0=[cos(meanTheta).*0.35,cos(meanTheta).*0.32,0.05,0];
oY0=[sin(meanTheta).*0.35,sin(meanTheta).*0.32,-0.15,0];
bXY0=bezierCurve([oX0',oY0'],200);
obj.ClassLine(i)=plot(bXY0(:,1),bXY0(:,2),'Color',[colorList(i,:),.7],'LineWidth',1.4);
obj.ClassScatter1(i)=scatter(cos(meanTheta).*0.35,sin(meanTheta).*0.35,'filled',...
'CData',colorList(i,:),'MarkerEdgeColor',[.4,.4,.4],'SizeData',40);
rotation=meanTheta/pi*180;
if rotation>90&&rotation<270
rotation=rotation+180;
obj.LabelHdl0(i)=text(cos(meanTheta).*0.35,sin(meanTheta).*0.35,classNameList{i},...
'Rotation',rotation,'HorizontalAlignment','right','FontSize',12,'VerticalAlignment','bottom');
else
obj.LabelHdl0(i)=text(cos(meanTheta).*0.35,sin(meanTheta).*0.35,classNameList{i},...
'Rotation',rotation,'FontSize',12,'VerticalAlignment','cap');
end
end
obj.MeanScatter=scatter(0,0,'filled','CData',mean(colorList(1:length(classNameList),:)),'MarkerEdgeColor',[.4,.4,.4],'SizeData',60);
disp(char([64 97 117 116 104 111 114 32 58 32 115 108 97 110 100 97 114 101 114]))
% 贝塞尔函数 ===================================================
function pnts=bezierCurve(pnts,N)
t=linspace(0,1,N);
p=size(pnts,1)-1;
coe1=factorial(p)./factorial(0:p)./factorial(p:-1:0);
coe2=((t).^((0:p)')).*((1-t).^((p:-1:0)'));
pnts=(pnts'*(coe1'.*coe2))';
end
end
% 修改颜色函数
function obj=setColorN(obj,n,color)
set(obj.ClassLine(n),'Color',[color,.8]);
set(obj.ChildLine1(n),'Color',[color,.4]);
for i=1:length(obj.ChildLine2(n,:))
if obj.ChildLine2(n,i)~=0
set(obj.ChildLine2(n,i),'Color',[color,.4]);
end
end
set(obj.ClassScatter1(n),'MarkerFaceColor',color);
set(obj.ClassScatter2(n),'CData',color);
tColorList=zeros(length(obj.ClassScatter1),3);
for i=1:length(obj.ClassScatter1)
tColorList(i,:)=get(obj.ClassScatter1(i),'CData');
end
set(obj.MeanScatter,'MarkerFaceColor',mean(tColorList));
set(obj.MeanScatter,'CData',mean(tColorList));
end
% 修改第一层文字属性
function obj=setLable1(obj,varargin)
for i=1:length(obj.LabelHdl0)
set(obj.LabelHdl0(i),varargin{:});
end
end
% 修改第二层文字属性
function obj=setLable2(obj,varargin)
for i=1:size(obj.LabelHdl1,1)
for j=1:size(obj.LabelHdl1,2)
if obj.LabelHdl1(i,j)~=0
set(obj.LabelHdl1(i,j),varargin{:});
end
end
end
end
end
% Zhaoxu Liu (2022). Circular dendrogram 环形树状图
% (https://www.mathworks.com/matlabcentral/fileexchange/118325-circular-dendrogram),
% MATLAB Central File Exchange. 检索来源 2022/9/29.
end
MATLAB树状图绘制能画成这样属实不易,如果有用请留个赞
叭~
未经允许本代码请勿作商业用途,引用的话可以引用我file exchange上的链接,可使用如下格式:
Zhaoxu Liu (2022). Circular dendrogram 环形树状图 (https://www.mathworks.com/matlabcentral/fileexchange/118325-circular-dendrogram), MATLAB Central File Exchange. 检索来源 2022/9/29.
若转载请保留以上file exchange链接及本文链接!!!
全部m文件及数据获取:
链接:https://pan.baidu.com/s/12iTl6P82PsMUhdZc4tEQJg?pwd=slan
提取码:slan