matlab官方库没有桑基图(sankey) 那就自己写一个

已于 2022-05-23 20:57:15 修改
阅读量3k 收藏 55
点赞数 17
分类专栏： # MATLAB 科学绘图 文章标签： MATLAB sankey 桑基图 Sankey Diagram 函数
于 2021-06-21 14:19:28 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/slandarer/article/details/118084972
版权

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

在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
先说明函数(sankey2)怎么用，函数完整代码放在博客最后

详细用法

1 使用示例

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

List={'a1',1,'A';
      'a2',1,'A';
      'a3',1,'A';
      'a3',0.5,'C';
      'b1',1,'B';
      'b2',1,'B';
      'b3',1,'B';
      'c1',1,'C';
      'c2',1,'C';
      'c3',1,'C';
      'A',2,'AA';
      'A',1,'BB';
      'B',1.5,'BB';
      'B',1.5,'AA';
      'C',3.5,'BB';
      };
axis([0,2,0,1])
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数组里：

List={'a1',1,'A';
      'a2',1,'A';
      'a3',1,'A';
      'a3',0.5,'C';
      'b1',1,'B';
      'b2',1,'B';
      'b3',1,'B';
      'c1',1,'C';
      'c2',1,'C';
      'c3',1,'C';
      'A',2,'AA';
      'A',1,'BB';
      'B',1.5,'BB';
      'B',1.5,'AA';
      'C',3.5,'BB';
      };

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

2.2 修饰参数

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

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

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

List={'a1',1,'A';
      'a2',1,'A';
      'a3',1,'A';
      'a3',0.5,'C';
      'b1',1,'B';
      'b2',1,'B';
      'b3',1,'B';
      'c1',1,'C';
      'c2',1,'C';
      'c3',1,'C';
      'A',2,'AA';
      'A',1,'BB';
      'B',1.5,'BB';
      'B',1.5,'AA';
      'C',3.5,'BB';
      };
axis([0,2,0,1])
colorList=[0.4600    0.5400    0.4600
    0.5400    0.6800    0.4600
    0.4100    0.4900    0.3600
    0.3800    0.5300    0.8400
    0.4400    0.5900    0.8700
    0.5800    0.7900    0.9300
    0.6500    0.6400    0.8400
    0.6300    0.6300    0.8000
    0.5600    0.5300    0.6700
    0.7600    0.8100    0.4300
    0.5600    0.8600    0.9700
    0.7800    0.5900    0.6500
    0.8900    0.9100    0.5300
    0.9300    0.5600    0.2500];
sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)

在这里插入图片描述

3 输出

函数的输出如下：
在这里插入图片描述
nameList就是每一个元素的名称，block是每一个方块的图形对象，connect是每一个连接的图形对象，txt是每一个标签的文本对象，我们可以做出以下操作：

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

List={'a1',1,'A';
      'a2',1,'A';
      'a3',1,'A';
      'a3',0.5,'C';
      'b1',1,'B';
      'b2',1,'B';
      'b3',1,'B';
      'c1',1,'C';
      'c2',1,'C';
      'c3',1,'C';
      'A',2,'AA';
      'A',1,'BB';
      'B',1.5,'BB';
      'B',1.5,'AA';
      'C',3.5,'BB';
      };
axis([0,2,0,1])
sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)

在这里插入图片描述

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

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

在这里插入图片描述
操作2 把第10个方块的第一个连接变为红色

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

在这里插入图片描述
操作3 把第11个标签文本放大

sankeyHdl.txt(11).FontSize=40;

在这里插入图片描述

函数完整代码

function sankeyHdl=sankey2(varargin)
if strcmp(get(varargin{1},'type'),'axes' )
    ax=varargin{1};
else
    ax=gca;
end
hold(ax,'on')

%若未设置，则图像的初始值==================================================
prop.Color=[0,0,0];
prop.FontSize=10;
prop.FontColor=[0,0,0];
prop.Xlim=[0,1];
prop.YLim=[0,1];
prop.PieceWidth=0.15;
prop.List=[];
prop.Margin=0.05;
prop.Sep=1/8;
prop.EdgeColor=[0 0 0];

%从可变长度变量中提取有用信息==============================================
for i=1:length(varargin)
    tempVar=varargin{i};
    if ischar(tempVar)&&length(tempVar)>1
        prop.(tempVar)=varargin{i+1};
    end
end

%流量矩阵构建==============================================================
nameList=unique([prop.List(:,1);prop.List(:,3)],'stable');
blockMat=zeros(length(nameList));
for i=1:size(prop.List,1)
    s=strcmp(nameList,prop.List(i,1));
    e=strcmp(nameList,prop.List(i,3));
    blockMat(s,e)=prop.List{i,2};
end
totalFlow=max([sum(blockMat,1);sum(blockMat,2)'],[],1);


%划分桑基图层次============================================================
List_L=prop.List(:,1);
List_R=prop.List(:,3);
prop.layer=[];layerRoot=[];n=1;
for i=length(List_R):-1:1
    if ~any(strcmp(List_L,List_R{i}))
        layerRoot=[layerRoot;find(strcmp(nameList,List_R{i}))];
    end
end
layerRoot=unique(layerRoot,'stable');
while ~isempty(List_L)
    layer_n=[];
    for i=length(List_L):-1:1
        if ~any(strcmp(List_R,List_L{i}))
            layer_n=[layer_n;find(strcmp(nameList,List_L{i}))];
            List_L(i)=[];
            List_R(i)=[];
        end
    end
    layer_n=unique(layer_n,'stable');
    prop.layer(length(layer_n),n)=0;
    prop.layer(1:length(layer_n),n)=layer_n;
    n=n+1;
end
prop.layer(length(layerRoot),n)=0;
prop.layer(1:length(layerRoot),n)=layerRoot;
prop.layerNum=size(prop.layer,2);




%绘制方块==================================================================
baseBlockX=[0,1,1,0];
baseBlockY=[0,0,1,1];
bnul=max(sum(prop.layer~=0,1));   %block number upper limit
baseLenY=(diff(prop.YLim)-2*prop.Margin)/(bnul+(bnul-1)*prop.Sep)*bnul;
baseLenX=(diff(prop.XLim)-2*prop.Margin)/(prop.layerNum-0.5);
colorIndex=1;
for i=1:prop.layerNum
    tempY=prop.Margin;
    elemSet=prop.layer(prop.layer(:,i)~=0,i);
    flowSet=totalFlow(elemSet);
    offSet=(diff(prop.YLim)-2*prop.Margin-baseLenY/length(elemSet)*((length(elemSet)+(length(elemSet)-1)*prop.Sep)))/2;
    for j=1:length(elemSet)
        tempLenY=baseLenY./sum(flowSet).*flowSet(j);
        
        sankeyHdl.block(prop.layer(j,i))=...
        fill(baseBlockX.*prop.PieceWidth+prop.Margin+(i-1)*baseLenX,...
            baseBlockY.*tempLenY+tempY+offSet,...
            prop.Color(colorIndex,:),'EdgeColor',prop.EdgeColor);
        
        tempY=tempY+tempLenY+baseLenY/length(elemSet)*prop.Sep;
        colorIndex=mod(colorIndex,size(prop.Color,1))+1;
    end
end

%绘制连接
layerList=prop.layer(:);
for i=1:length(nameList)
    for j=i:length(nameList)
        if blockMat(i,j)~=0
            Hdl_L=sankeyHdl.block(i);
            Hdl_R=sankeyHdl.block(j);
            list_L=find(blockMat(i,:)~=0);
            list_R=find(blockMat(:,j)~=0);
            [~,pl,~]=intersect(layerList,list_L(:));
            [~,pr,~]=intersect(layerList,list_R(:));
            list_L=layerList(sort(pl));
            list_R=layerList(sort(pr));
            flow_L=blockMat(i,list_L);
            flow_R=blockMat(list_R,j);
            XData_L=Hdl_L.XData;YData_L=Hdl_L.YData;
            XData_R=Hdl_R.XData;YData_R=Hdl_R.YData;
            xx=[XData_L(1:2);XData_R(1:2)]';
            k_L=find(list_L==j);
            k_R=find(list_R==i);
            yy=[YData_L(1:2)+(YData_L(3:4)-YData_L(1:2))./sum(flow_L).*sum(flow_L(1:k_L-1));
                YData_R(1:2)+(YData_R(3:4)-YData_R(1:2))./sum(flow_R).*sum(flow_R(1:k_R-1))]';
            xxq=XData_L(2):0.01:XData_R(1);
            yyq=interp1(xx,yy,xxq,'pchip');
            tempColor=Hdl_L.FaceColor;
            width=(YData_R(3)-YData_R(1))./sum(flow_R).*flow_R(k_R);
             sankeyHdl.connect(i,k_L)=...
            fill([xxq,xxq(end:-1:1)],[yyq,yyq(end:-1:1)+width],tempColor,'EdgeColor','none','FaceAlpha',0.3);
        end    
    end
end

%绘制文本
for i=1:prop.layerNum
    tempY=prop.Margin;
    elemSet=prop.layer(prop.layer(:,i)~=0,i);
    flowSet=totalFlow(elemSet);
    offSet=(diff(prop.YLim)-2*prop.Margin-baseLenY/length(elemSet)*((length(elemSet)+(length(elemSet)-1)*prop.Sep)))/2;
    for j=1:length(elemSet)
        tempLenY=baseLenY./sum(flowSet).*flowSet(j);
        
        sankeyHdl.txt(prop.layer(j,i))=...
        text(prop.PieceWidth+prop.Margin+(i-1)*baseLenX,tempLenY/2+tempY+offSet,[' ',nameList{elemSet(j)}],...
            'FontSize',prop.FontSize,'Color',prop.FontColor);
        
        tempY=tempY+tempLenY+baseLenY/length(elemSet)*prop.Sep;
    end
end
sankeyHdl.nameList=nameList';
end

使用示例代码

List={'零食',300,'食品开销';
      '饮料',150,'食品开销';
      '水果',250,'食品开销';
      '食堂',600,'食品开销';
      '外卖',400,'食品开销';
      '水费',90,'住宿开销';
      '电费',90,'住宿开销';
      '网费',120,'住宿开销';
      '食品开销',1700,'开销';
      '住宿开销',300,'开销'};
axis([0,2,0,1])
colorList=[0.4600    0.5400    0.4600
    0.5400    0.6800    0.4600
    0.4100    0.4900    0.3600
    0.3800    0.5300    0.8400
    0.4400    0.5900    0.8700
    0.5800    0.7900    0.9300
    0.6500    0.6400    0.8400
    0.6300    0.6300    0.8000
    0.5600    0.5300    0.6700
    0.7600    0.8100    0.4300
    0.5600    0.8600    0.9700
    0.7800    0.5900    0.6500
    0.8900    0.9100    0.5300
    0.9300    0.5600    0.2500];
sankeyHdl=sankey2([],'XLim',[0,2],'YLim',[0,1],'PieceWidth',0.15,'List',List,'Color',colorList)

sankeyHdl.block(4).FaceColor=[0.8,0.3,0.3];
sankeyHdl.txt(4).FontSize=40;

在这里插入图片描述
试试证明还是方块画细一点显得高级，修饰一下：
在这里插入图片描述
在这里插入图片描述

slandarer
关注
专栏目录
数据可视化之 Sankey 桑基图的实现
weixin_33748818的博客
12-24 9148
原文地址：https://geekplux.com/2018/08/28/how-to-implement-sankey-diagram.html 什么是桑基图 Google 搜索桑基图，可以搜到一大堆定义。简而言之，桑基图是一种数据流图，展示了数据是如何从左到右流向最后的节点，每条边代表一条数据流，宽度代表数据流的大小。桑基图常用于...
桑基图
weixin_40903057的博客
07-10 1173
Sankey（桑基图） 桑基图是一种特殊的流图, 它主要用来表示原材料、能量等如何从初始形式经过中间过程的加工、转化到达最终形式。 # 导入桑基图类 from pyecharts import Sankey nodes = [ {'name': '自己申请'}, {'name': '朋友推荐'}, {'name': '参加招聘会'}, {'name': '上海'}, {'name...
MATLAB | 绘图复刻(十二) | 桑基图+气泡图
slandarer的博客
09-12 1685
~
drawSankey:drawSankey 是一个 MatLab 函数，用于绘制单向桑基图-matlab开发
06-01
drawSankey 是一个 matlab 函数，可以绘制单向 Sankey 图（即没有反馈回路），但是可以指定多个输入。 语法如下： drawSankey（输入，损失，单位，标签，sep） 输入：包含流输入的向量，其中第一个将被视为主要输入并集中绘制，其他输入将显示在此下方。 损失：包含系统所有损失的向量，将显示在桑基图的顶部 unit：一个字符串，表示流的表达单位 标签：不同流的标签的单元格列表，从输入的标签开始，然后是损失，最后是输出 sep：用于分隔线的位置列表（可选），放置在与提供的索引相对应的损失之后 当前版本：02.11.2009 开发者：James SPELLING, KTH-EGI-EKV 根据知识共享署名-非商业性使用-相同方式共享 (by-nc-sa) 分发，被许可人仅可出于非商业目的复制、分发、展示和执行该作品以及基于该作品制作衍生作品。 使用、分发或修改
桑基图：创建一个包含多个块的桑基-matlab开发
05-29
该代码主要基于冲积图 ( https://nl.mathworks.com/matlabcentral/fileexchange/66746-alluvial-flow-diagram ) 创建。 它在两个方面改进了冲积图： 1. 可以绘制多块Sankey图2.对于每个块，左右两端不必平衡。 文本标签必须单独添加。
桑基图 Matlab
matlab_dingdang的博客
12-11 1004
桑基图是一种用于描述复杂系统的图形化工具，它可以帮助人们更好地理解系统中各个因素之间的关系。桑基图最初是由美国工程师马修·桑基在1898年提出的，他在一篇名为《桑基图及其应用》的论文中首次介绍了这种图表。自那时起，桑基图已经被广泛应用于各个领域，包括工程、管理、经济学和环境科学等。桑基图通常由一系列水平和垂直的条形组成，每个条形代表系统中的一个因素或变量。这些条形之间的宽度表示不同因素之间的关系强度或者数量的差异。
MATLAB | 如何使用MATLAB绘制高度自定义的桑基图(sankey)
slandarer的博客
04-28 3442
~
matlab开发-drawSankey
08-22
matlab开发-drawSankey。drawsankey是一个绘制单向sankey图的matlab函数。
matlab绘桑基图
08-21
在 MATLAB 中绘制桑基图（Sankey diagram），你可以使用 MathWorks 提供的 Sankey 函数库。以下是一个简单的示例代码： ```matlab % 创建节点和连接的数据 nodes = {'A', 'B', 'C', 'D', 'E'}; % 节点名称 links = ...
matlab怎么画桑基图
05-24
Matlab中可以使用sangraph函数来绘制桑基图，以下是一个简单的示例代码： ```matlab % 创建一个随机的桑基图数据 nodes = {'A', 'B', 'C', 'D', 'E', 'F'}; values = randi([10, 50], 6, 2); % 绘制桑基图 sankey...
d3.js 桑基图（sankey）插件
05-10
d3.js 桑基图（sankey）插件
Matlab绘制桑基图的代码
11-12
Matlab绘制桑基图的代码
pySankey:使用matplotlib创建sankey图
05-22
pySankey 使用matplotlib创建仅从左向右流动的简单。 要求 需要python-tk（对于python 2.7）或python3-tk（对于python 3.x），您可以使用以下命令安装其他要求： pip install -r requirements.txt 例子 随着fruits.txt： 真的 预料到的 0 蓝莓 橙子 1个 酸橙 橙子 2个 蓝莓 酸橙 3 苹果 橙子 ... ... ... 996 酸橙 橙子 997 蓝莓 橙子 998 橙子 香蕉 999 苹果 酸橙 1000行×2列 您可以使用以下代码生成sankey的图： import pandas as pd from pysankey import sankey pd . options . display . max_rows = 8 df = pd . read_cs
pySankey2：使用matplotlib的静态Sankey图
02-05
pySankey2 pySankey2开发用于基于matplotlib绘制静态Sankey图。 的原始版本是由anazalea等开发的。 ，此版本仅支持2层Sankey图，并且可设置的参数受到限制。 pySankey2扩展了原始版本的功能，以支持多层Sankey图，并允许用户更灵活地设置参数。 安装 v0.1.0 pysankey2-v0.1.0在可用，下载后可以通过pip cmd在本地安装： pip install pysankey2-0.1.0.zip 开发版 pip install git+https://github.com/SZJShuffle/pySankey2.git 用
Matlab进阶绘图第31期—桑基图（Sankey Chart）
qq_26447137的博客
10-16 464
Matlab进阶绘图第31期—桑基图（Sankey Chart）
MATLAB | 特殊绘图小合集(二) | 金字塔图、类桑基图、y轴分段填充折线图
slandarer的博客
02-21 1281
~
MATLAB | 绘图复刻(十四) | 右侧对齐桑基图，及工具函数SSankey更新
最新发布
slandarer的博客
02-04 1628
~
桑基图（sankey）
weixin_30394981的博客
04-26 814
一、桑基图简介 　　桑基图是一种特殊的流图, 它主要用来表示原材料、能量等如何从初始形式经过中间过程的加工、转化到达最终形式，如下所示为最基本的事物状态随时间推移的变化。桑基图的type为sankey， 它是echart3中新出现的类型，需要加载echart3类库才能实现。 　　 二、桑基图制作 1、如下是上图的前端实现： 1 var statisChangeCh...
Matplotlib笔记4 Sankey图
yangguangdblu的博客
12-02 1554
import matplotlib.pyplot as plt #plt.rcParams['font.sans-serif']=['SimHei'] #用来正常显示中文标签 #plt.rcParams['axes.unicode_minus']=False #用来正常显示负号 # 有中文出现的情况，需要u'内容' import numpy as np from matplotlib.sankey...
评论 18
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

slandarer

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值