经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置：技术经验 » 程序设计 » MATLAB » 查看文章
详解Matlab如何绘制桑基图
来源：jb51　　时间：2022/3/2 13:43:17　　对本文有异议

这次主要是分享自己写的一个函数，用来绘制桑基图，效果大概是下面这样子：

先说明函数(sankey2)怎么用，函数完整代码放在博客最后

详细用法

1 使用示例

新建一个m文件，运行如下代码

  1. List={'a1',1,'A';
  2.       'a2',1,'A';
  3.       'a3',1,'A';
  4.       'a3',0.5,'C';
  5.       'b1',1,'B';
  6.       'b2',1,'B';
  7.       'b3',1,'B';
  8.       'c1',1,'C';
  9.       'c2',1,'C';
  10.       'c3',1,'C';
  11.       'A',2,'AA';
  12.       'A',1,'BB';
  13.       'B',1.5,'BB';
  14.       'B',1.5,'AA';
  15.       'C',3.5,'BB';
  16.       };
  17. axis([0,2,0,1])
  18. sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',[0.3,0.3,0.7])

2 输入参数

2.1 必要输入参数

必要参数一共有四个，第一个就是流量表(‘List’)，可以从excel里导入，也可以像下图这样直接写在cell数组里：

  1. List={'a1',1,'A';
  2.       'a2',1,'A';
  3.       'a3',1,'A';
  4.       'a3',0.5,'C';
  5.       'b1',1,'B';
  6.       'b2',1,'B';
  7.       'b3',1,'B';
  8.       'c1',1,'C';
  9.       'c2',1,'C';
  10.       'c3',1,'C';
  11.       'A',2,'AA';
  12.       'A',1,'BB';
  13.       'B',1.5,'BB';
  14.       'B',1.5,'AA';
  15.       'C',3.5,'BB';
  16.       };

另外三个参数为，x轴范围(‘xLim’),y轴范围(‘YLim’),以及函数第一个参数，要绘制图像的axes，要画在当前窗口可以将第一个参数写作[]或者gca

2.2 修饰参数

主要有以下几个，不写时用省缺值替换

  • Color 方块颜色
  • EdgeColor 方块边缘颜色
  • FontSize 字号
  • FontColor 字体颜色
  • PieceWidth 方块宽度
  • Margin 桑基图距离边缘距离
  • Sep 纵向空白占方块总长度比例

其中Color可以是个nx3大小的矩阵，矩阵中数值范围为[0,1]，例如：

  1. List={'a1',1,'A';
  2.       'a2',1,'A';
  3.       'a3',1,'A';
  4.       'a3',0.5,'C';
  5.       'b1',1,'B';
  6.       'b2',1,'B';
  7.       'b3',1,'B';
  8.       'c1',1,'C';
  9.       'c2',1,'C';
  10.       'c3',1,'C';
  11.       'A',2,'AA';
  12.       'A',1,'BB';
  13.       'B',1.5,'BB';
  14.       'B',1.5,'AA';
  15.       'C',3.5,'BB';
  16.       };
  17. axis([0,2,0,1])
  18. colorList=[0.4600    0.5400    0.4600
  19.     0.5400    0.6800    0.4600
  20.     0.4100    0.4900    0.3600
  21.     0.3800    0.5300    0.8400
  22.     0.4400    0.5900    0.8700
  23.     0.5800    0.7900    0.9300
  24.     0.6500    0.6400    0.8400
  25.     0.6300    0.6300    0.8000
  26.     0.5600    0.5300    0.6700
  27.     0.7600    0.8100    0.4300
  28.     0.5600    0.8600    0.9700
  29.     0.7800    0.5900    0.6500
  30.     0.8900    0.9100    0.5300
  31.     0.9300    0.5600    0.2500];
  32. sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)

3 输出

函数的输出如下：

nameList就是每一个元素的名称，block是每一个方块的图形对象，connect是每一个连接的图形对象，txt是每一个标签的文本对象，我们可以做出以下操作：

图形对象和文本对象原本自带的属性依旧可以用(颜色，位置，透明度，边缘粗细等等)
假设我们编写了如下代码的基础上，我们尝试对对象进行操作：

  1. List={'a1',1,'A';
  2.       'a2',1,'A';
  3.       'a3',1,'A';
  4.       'a3',0.5,'C';
  5.       'b1',1,'B';
  6.       'b2',1,'B';
  7.       'b3',1,'B';
  8.       'c1',1,'C';
  9.       'c2',1,'C';
  10.       'c3',1,'C';
  11.       'A',2,'AA';
  12.       'A',1,'BB';
  13.       'B',1.5,'BB';
  14.       'B',1.5,'AA';
  15.       'C',3.5,'BB';
  16.       };
  17. axis([0,2,0,1])
  18. sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)

操作1 把第四个方块颜色变为红色

  1. sankeyHdl.block(4).FaceColor=[0.8,0.3,0.3];

操作2 把第10个方块的第一个连接变为红色

  1. sankeyHdl.connect(10,1).FaceColor=[0.8,0.3,0.3];

操作3 把第11个标签文本放大

  1. sankeyHdl.txt(11).FontSize=40;

函数完整代码

  1. function sankeyHdl=sankey2(varargin)
  2. if strcmp(get(varargin{1},'type'),'axes' )
  3.     ax=varargin{1};
  4. else
  5.     ax=gca;
  6. end
  7. hold(ax,'on')
  8.  
  9. %若未设置，则图像的初始值==================================================
  10. prop.Color=[0,0,0];
  11. prop.FontSize=10;
  12. prop.FontColor=[0,0,0];
  13. prop.Xlim=[0,1];
  14. prop.YLim=[0,1];
  15. prop.PieceWidth=0.15;
  16. prop.List=[];
  17. prop.Margin=0.05;
  18. prop.Sep=1/8;
  19. prop.EdgeColor=[0 0 0];
  20.  
  21. %从可变长度变量中提取有用信息==============================================
  22. for i=1:length(varargin)
  23.     tempVar=varargin{i};
  24.     if ischar(tempVar)&&length(tempVar)>1
  25.         prop.(tempVar)=varargin{i+1};
  26.     end
  27. end
  28.  
  29. %流量矩阵构建==============================================================
  30. nameList=unique([prop.List(:,1);prop.List(:,3)],'stable');
  31. blockMat=zeros(length(nameList));
  32. for i=1:size(prop.List,1)
  33.     s=strcmp(nameList,prop.List(i,1));
  34.     e=strcmp(nameList,prop.List(i,3));
  35.     blockMat(s,e)=prop.List{i,2};
  36. end
  37. totalFlow=max([sum(blockMat,1);sum(blockMat,2)'],[],1);
  38.  
  39.  
  40. %划分桑基图层次============================================================
  41. List_L=prop.List(:,1);
  42. List_R=prop.List(:,3);
  43. prop.layer=[];layerRoot=[];n=1;
  44. for i=length(List_R):-1:1
  45.     if ~any(strcmp(List_L,List_R{i}))
  46.         layerRoot=[layerRoot;find(strcmp(nameList,List_R{i}))];
  47.     end
  48. end
  49. layerRoot=unique(layerRoot,'stable');
  50. while ~isempty(List_L)
  51.     layer_n=[];
  52.     for i=length(List_L):-1:1
  53.         if ~any(strcmp(List_R,List_L{i}))
  54.             layer_n=[layer_n;find(strcmp(nameList,List_L{i}))];
  55.             List_L(i)=[];
  56.             List_R(i)=[];
  57.         end
  58.     end
  59.     layer_n=unique(layer_n,'stable');
  60.     prop.layer(length(layer_n),n)=0;
  61.     prop.layer(1:length(layer_n),n)=layer_n;
  62.     n=n+1;
  63. end
  64. prop.layer(length(layerRoot),n)=0;
  65. prop.layer(1:length(layerRoot),n)=layerRoot;
  66. prop.layerNum=size(prop.layer,2);
  67.  
  68.  
  69.  
  70.  
  71. %绘制方块==================================================================
  72. baseBlockX=[0,1,1,0];
  73. baseBlockY=[0,0,1,1];
  74. bnul=max(sum(prop.layer~=0,1));   %block number upper limit
  75. baseLenY=(diff(prop.YLim)-2*prop.Margin)/(bnul+(bnul-1)*prop.Sep)*bnul;
  76. baseLenX=(diff(prop.XLim)-2*prop.Margin)/(prop.layerNum-0.5);
  77. colorIndex=1;
  78. for i=1:prop.layerNum
  79.     tempY=prop.Margin;
  80.     elemSet=prop.layer(prop.layer(:,i)~=0,i);
  81.     flowSet=totalFlow(elemSet);
  82.     offSet=(diff(prop.YLim)-2*prop.Margin-baseLenY/length(elemSet)*((length(elemSet)+(length(elemSet)-1)*prop.Sep)))/2;
  83.     for j=1:length(elemSet)
  84.         tempLenY=baseLenY./sum(flowSet).*flowSet(j);
  85.         
  86.         sankeyHdl.block(prop.layer(j,i))=...
  87.         fill(baseBlockX.*prop.PieceWidth+prop.Margin+(i-1)*baseLenX,...
  88.             baseBlockY.*tempLenY+tempY+offSet,...
  89.             prop.Color(colorIndex,:),'EdgeColor',prop.EdgeColor);
  90.         
  91.         tempY=tempY+tempLenY+baseLenY/length(elemSet)*prop.Sep;
  92.         colorIndex=mod(colorIndex,size(prop.Color,1))+1;
  93.     end
  94. end
  95.  
  96. %绘制连接
  97. layerList=prop.layer(:);
  98. for i=1:length(nameList)
  99.     for j=i:length(nameList)
  100.         if blockMat(i,j)~=0
  101.             Hdl_L=sankeyHdl.block(i);
  102.             Hdl_R=sankeyHdl.block(j);
  103.             list_L=find(blockMat(i,:)~=0);
  104.             list_R=find(blockMat(:,j)~=0);
  105.             [~,pl,~]=intersect(layerList,list_L(:));
  106.             [~,pr,~]=intersect(layerList,list_R(:));
  107.             list_L=layerList(sort(pl));
  108.             list_R=layerList(sort(pr));
  109.             flow_L=blockMat(i,list_L);
  110.             flow_R=blockMat(list_R,j);
  111.             XData_L=Hdl_L.XData;YData_L=Hdl_L.YData;
  112.             XData_R=Hdl_R.XData;YData_R=Hdl_R.YData;
  113.             xx=[XData_L(1:2);XData_R(1:2)]';
  114.             k_L=find(list_L==j);
  115.             k_R=find(list_R==i);
  116.             yy=[YData_L(1:2)+(YData_L(3:4)-YData_L(1:2))./sum(flow_L).*sum(flow_L(1:k_L-1));
  117.                 YData_R(1:2)+(YData_R(3:4)-YData_R(1:2))./sum(flow_R).*sum(flow_R(1:k_R-1))]';
  118.             xxq=XData_L(2):0.01:XData_R(1);
  119.             yyq=interp1(xx,yy,xxq,'pchip');
  120.             tempColor=Hdl_L.FaceColor;
  121.             width=(YData_R(3)-YData_R(1))./sum(flow_R).*flow_R(k_R);
  122.              sankeyHdl.connect(i,k_L)=...
  123.             fill([xxq,xxq(end:-1:1)],[yyq,yyq(end:-1:1)+width],tempColor,'EdgeColor','none','FaceAlpha',0.3);
  124.         end    
  125.     end
  126. end
  127.  
  128. %绘制文本
  129. for i=1:prop.layerNum
  130.     tempY=prop.Margin;
  131.     elemSet=prop.layer(prop.layer(:,i)~=0,i);
  132.     flowSet=totalFlow(elemSet);
  133.     offSet=(diff(prop.YLim)-2*prop.Margin-baseLenY/length(elemSet)*((length(elemSet)+(length(elemSet)-1)*prop.Sep)))/2;
  134.     for j=1:length(elemSet)
  135.         tempLenY=baseLenY./sum(flowSet).*flowSet(j);
  136.         
  137.         sankeyHdl.txt(prop.layer(j,i))=...
  138.         text(prop.PieceWidth+prop.Margin+(i-1)*baseLenX,tempLenY/2+tempY+offSet,[' ',nameList{elemSet(j)}],...
  139.             'FontSize',prop.FontSize,'Color',prop.FontColor);
  140.         
  141.         tempY=tempY+tempLenY+baseLenY/length(elemSet)*prop.Sep;
  142.     end
  143. end
  144. sankeyHdl.nameList=nameList';
  145. end

使用示例代码

  1. List={'零食',300,'食品开销';
  2.       '饮料',150,'食品开销';
  3.       '水果',250,'食品开销';
  4.       '食堂',600,'食品开销';
  5.       '外卖',400,'食品开销';
  6.       '水费',90,'住宿开销';
  7.       '电费',90,'住宿开销';
  8.       '网费',120,'住宿开销';
  9.       '食品开销',1700,'开销';
  10.       '住宿开销',300,'开销'};
  11. axis([0,2,0,1])
  12. sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)
  13.  
  14. sankeyHdl.block(4).FaceColor=[0.8,0.3,0.3];
  15. sankeyHdl.txt(4).FontSize=40;

试试证明还是方块画细一点显得高级，修饰一下：

以上就是详解Matlab如何绘制桑基图的详细内容，更多关于Matlab桑基图的资料请关注w3xue其它相关文章！

 友情链接：直通硅谷 　点职佳 　北美留学生论坛

MATLAB热门文章

  • Matlab的solve（）函数的使用方法
  • Matlab绘图——基于CST导出数据（CST图片Matlab实现）
  • Matlab分布云图绘制（渐变彩色）
  • MATLAB最大均值差异(Maximum Mean Discrepancy)
  • 利用matlab写一个简单的拉普拉斯变换提取图像边缘
  • 密度峰值聚类算法MATLAB程序
  • 详解Matlab如何绘制桑基图
  • MATLAB中“fitgmdist”的用法及其GMM聚类算法
  • python调用matlab的方法详解
  • 相机标定问题-Matlab & Py-Opencv

    • MATLAB推荐文章

  • 使用Matlab绘制七夕咕呱小青蛙
  • Matlab实现极坐标堆叠柱状图的绘制
  • 基于Matlab实现水波倒影特效的制作
  • 教你用Matlab制作黄金矿工小游戏
  • 利用Matlab复刻举牌加油小人生成器
  • 详解Matlab如何绘制桑基图
  • 教你用Matlab制作立体动态相册
  • 详解Matlab绘制3D玫瑰花的方法(内附旋转版本)
  • 详解基于Matlab的空心散点检测
  • 教你使用Matlab制作图形验证码生成器(app designer)

    • 本站QQ群：前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

    W3xue 的所有内容仅供测试，对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
    关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接（目前9元/月）请联系QQ：27243702 沸活量
    皖ICP备17017327号-2　皖公网安备34020702000426号