MATLAB | Как использовать MATLAB для рисования индивидуально настроенной диаграммы Санки (sankey)

Language 2023-05-07 20:18:28 views: null

Я также создал очень простую функцию рисования диаграммы Санки раньше, но она не может справиться со многими особыми случаями.Здесь я провел рефакторинг некоторых из них и записал в класс, добавив множество встроенных функций оформления, чтобы понять, что входящие и исходящие данные не равны или Рисунок частных случаев межуровневого потока данных, давайте сначала покажем, какой эффект может быть достигнут с помощью написанной мной функции:

Приведенные выше изображения являются примерами этой статьи, давайте сначала объясним, как ее использовать, и поместим функции инструмента в конец статьи! ! !

Введение в использование

0 ввод данных

Для ввода данных требуется массив ячеек из трех столбцов, которые соответственно представляют начальную точку исходного направления потока, конечную точку целевого направления потока, а значение представляет собой объем передаваемых данных:

links={
    
    'a1','A',1.2;
       'a2','A',1;
       'a1','B',.6;
       'a3','A',1; 
       'a3','C',.5;
       'b1','B',.4; 
       'b2','B',1;
       'b3','B',1; 
       'c1','C',1;
       'c2','C',1;  
       'c3','C',1;
       'A','AA',2; 
       'A','BB',1.2;
       'B','BB',1.5; 
       'B','AA',1.5; 
       'C','BB',2.3; 
       'C','AA',1.2};

1 базовый чертеж

На самом деле, в основном нужно рисовать две линии:

links={
    
    'a1','A',1.2;'a2','A',1;'a1','B',.6;'a3','A',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','AA',1.5; 'C','BB',2.3; 'C','AA',1.2};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 开始绘图(Start drawing)
SK.draw()

2 Метод рендеринга соединения

Методы рендеринга соединения:

  • 'левый'
  • 'верно'
  • 'интерп' (по умолчанию)
  • 'карта'
  • 'простой'

RenderingMethodСреди них интерпретация рендеринга градиента является методом рендеринга по умолчанию, и метод рендеринга можно изменить, например, установив атрибуты перед draw (или предыдущими данными):

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改链接颜色渲染方式(Set link color rendering method)
% 'left'/'right'/'interp'(default)/'map'/'simple'
SK.RenderingMethod='left';  

% 开始绘图(Start drawing)
SK.draw()

левый

То есть цвет соединения такой же, как у левого квадрата узла:

верно

интерп

карта

То есть окрашивание выполняется в соответствии со значением, а цветовую полосу можно изменить через функцию палитры.

простой

Чистый серый:

выравнивание по 3 блокам

Он предназначен для установки выравнивания по верхнему краю, выравнивания по нижнему краю и центрирования «вверху»/«внизу»/«центре» (по умолчанию) соответственно и настраивается с помощью настроек. Его все равно необходимо установить перед рисованием Align:

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改对齐方式(Set alignment)
% 'up'/'down'/'center'(default)
SK.Align='up';

% 开始绘图(Start drawing)
SK.draw()

вверх

вниз

центр

4 текстовая позиция

Позиция текста — это не что иное, как выравнивание по центру вверх, вниз, влево и вправо: 'слева' (по умолчанию)/'справа'/'сверху'/'по центру'/'снизу', задается через атрибуты LabelLocation:

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改文本位置(Set Text Location)
% 'left'(default)/'right'/'top'/'center'/'bottom'
SK.LabelLocation='top';

SK.Sep=.2;

% 开始绘图(Start drawing)
SK.draw()

Обратите внимание, что когда положение установлено сверху и снизу, во избежание окклюзии вы можете установить Sepи настроить пропорцию пространства между блоками:

левый

верно

вершина

центр

нижний

5 Зазор и пропорция

Установив свойство, можно установить пропорцию ширины квадрата по горизонтали больше 0 и меньше 1. BlockScaleЧем больше пропорция, тем больше пропорция квадрата.

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 设置方块占比(Set the scale of blocks)
% BlockScale>0 & BlockScale<1
SK.BlockScale=.4;

% 开始绘图(Start drawing)
SK.draw()

Если скорректировано на 0,001:

При этом отрегулируйте соотношение верхнего и нижнего зазоров:

% 设置缝隙占比(Separation distance proportion)
SK.Sep=.4;

6 цветовая настройка

Измените предустановленный цвет , установив ColorListсвойства:

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 设置颜色(Set color)
SK.ColorList=[0.46, 0.54, 0.46;
    0.54, 0.68, 0.46;
    0.41, 0.49, 0.36;
    0.38, 0.53, 0.84;
    0.44, 0.59, 0.87;
    0.58, 0.79, 0.93;
    0.65, 0.64, 0.84;
    0.63, 0.63, 0.80;
    0.56, 0.53, 0.67;
    0.76, 0.81, 0.43;
    0.56, 0.86, 0.97;
    0.78, 0.59, 0.65;
    0.89, 0.91, 0.53;
    0.93, 0.56, 0.25;];

% 开始绘图(Start drawing)
SK.draw()

Конечно, вы можете изменить только часть цветовой схемы (например, изменить a3 на черный):

links={
    
    'a1','A',1.2;'a2','A',1;'a1','B',.6;'a3','A',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','AA',1.5; 'C','BB',2.3; 'C','AA',1.2};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 设置颜色(Set color)
SK.ColorList(3,:)=[0,0,0];

% 开始绘图(Start drawing)
SK.draw()

7 Входящие и исходящие данные не равны

Данные притока и оттока не равны, поэтому можно также построить специальные графики Исходные данные изменены с C на BB на 0,5 (ссылки{16,3}=.5):

% 流入及流出数据不相等(Unequal inflow and outflow data)
links={
    
    'a1','A',1.2;'a2','A',1;'a1','B',.6;'a3','A',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','AA',1.5; 'C','BB',2.3; 'C','AA',1.2};
links{
    
    16,3}=.5;

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 开始绘图(Start drawing)
SK.draw();

8 Поток через уровни

% 含跨层级流动(Including cross level flow)
links={
    
    'a1','A',1.2;'a2','A',2;'a1','B',.6;'a3','D',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','D',1.5; 'C','BB',2.3; 'C','AA',1.2;
       'D','AA',1.4; 'D','BB',1.1};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

SK.Sep=.1;

% 开始绘图(Start drawing)
SK.draw()

Таким образом, при прямом рисовании будет окклюзия, и будет довольно неудобная последовательность соединений (соединение внизу идет наверх, а соединение вверху идет на низ)

Порядок подключения по умолчанию

unique([Source;Target],'stable')

Рассчитывается таким образом, конечно, это свойство можно задать самому, предварительно настроим общий порядок соединений:

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改节点排列次序(Modify node arrangement order)
SK.NodeList={
    
    'a3','a1','a2','b1','b2','b3','c1','c2','c3','D','A','B','C','AA','BB'};

SK.Sep=.1;

% 开始绘图(Start drawing)
SK.draw()

Он более приятен для глаз, но он все еще заблокирован в середине.Если мы хотим переместить его вверх, мы можем использовать функцию moveBlockY, где узел D является 10-м узлом, и мы переместим его вверх на 6 единиц. значения (если значение велико, просто переместите больше), обратите внимание, что, поскольку объект, который нужно переместить, — это то, что уже было нарисовано, поэтому операцию перемещения следует размещать после операции рисования:

% 含跨层级流动(Including cross level flow)
links={
    
    'a1','A',1.2;'a2','A',2;'a1','B',.6;'a3','D',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','D',1.5; 'C','BB',2.3; 'C','AA',1.2;
       'D','AA',1.4; 'D','BB',1.1};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改节点排列次序(Modify node arrangement order)
SK.NodeList={
    
    'a3','a1','a2','b1','b2','b3','c1','c2','c3','D','A','B','C','AA','BB'};

SK.Sep=.1;

% 开始绘图(Start drawing)
SK.draw()

% 修改节点Y轴位置变化(Modify the position change of node Y direction)
SK.moveBlockY(10,+6);

9 Нарисовано графическое оформление

Чтобы изменить блок, используйте форму setBlock(n,prop,…), например (добавьте черный толстый контур к блоку № 2):

Поскольку то, что нужно изменить, это то, что уже было нарисовано, поэтому операцию модификации следует разместить после операции рисования:

links={
    
    'a1','A',1.2;'a2','A',1;'a1','B',.6;'a3','A',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','AA',1.5; 'C','BB',2.3; 'C','AA',1.2};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 开始绘图(Start drawing)
SK.draw()

% 设置方块属性(Set Block Properties)
SK.setBlock(2,'EdgeColor',[0,0,0],'LineWidth',6)

Измените цикл (все настроено на серый):

% 循环设置方块属性(Loop Set Block Properties)
for i=1:14
    SK.setBlock(i,'FaceColor',[.5,.5,.5])
end

Чтобы установить строку ссылки, используйте setLink(n,prop,…)

% 设置连接属性(Set Link Properties)
SK.setLink(5,'FaceColor',[0,0,0],'FaceAlpha',.5)

установить текст setLabel(n,prop,...)

% 设置标签属性(Set Label Properties)
SK.setLabel(11,'FontSize',40,'Color',[0,0,.8])

Кстати, просто добавьте заголовок или функцию title:

title(gca,'sankey plot by slandarer','FontSize',30,'FontName','Cambria')

10 Кольцевая диаграмма Санки

Функцию этой статьи можно связать с функцией ориентированного графа, которую я написал для создания кругового графа Санки.Функция ориентированного графа biChordChartможет быть получена на файловом обмене или складе gitee, и длина задачи не будет повторяться:

Обмен файлами направленного графа аккордов biChordChart

Zhaoxu Liu / slandarer (2023).Диаграмма аккордов орграфов ( https://www.mathworks.com/matlabcentral/fileexchange/121043-digraph-chord-chart ), MATLAB Central File Exchange. Получено с 2023/4/1.

склад

https://gitee.com/slandarer/matlab-аккорд-диаграмма

Нелицензионные коды нельзя использовать в каких-либо коммерческих целях, ссылка может ссылаться на указанную выше ссылку обмена файлами, а формат текста ссылки может быть изменен при цитировании, но по крайней мере ссылка должна быть включена.

Код использования связи

links={
    
    'a1','A',1.2;'a2','A',1;'a1','B',.6;'a3','A',1; 'a3','C',0.5;
       'b1','B',.4; 'b2','B',1;'b3','B',1; 'c1','C',1;
       'c2','C',1;  'c3','C',1;'A','AA',2; 'A','BB',1.2;
       'B','BB',1.5; 'B','AA',1.5; 'C','BB',2.3; 'C','AA',1.2};

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));
SK.draw();close all

figure('Name','sankey demo6','Units','normalized','Position',[.05,.05,.59,.8])
BCC=biChordChart(SK.AdjMat,'Arrow','on','Label',SK.NodeList);
BCC.CData=[[65,140,240;252,180,65;224,64,10;5,100,146;191,191,191;26,59,105;255,227,130;18,156,221;
    202,107,75;0,92,219;243,210,136;80,99,129;241,185,168;224,131,10;120,147,190]./255;
    [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];
BCC=BCC.draw();

% 添加刻度
BCC.tickState('on')
BCC.tickLabelState('on')

BCC.setTickFont('FontName','Cambria','FontSize',11)
BCC.setFont('FontName','Cambria','FontSize',17)

BCC.setLabelRadius(1.32);

11 Дополнительный маленький кейс (1)

Следует отметить, что узел здесь на китайском языке, и шрифт должен быть установлен на шрифт, поддерживающий китайский язык, например Song Ti.

clc;clear;
links{
    
    7,3}='';
for i=1:7
    links{
    
    i,1}=['浏览',num2str(i)];
    links{
    
    i,2}=['浏览',num2str(i+1)];
    links{
    
    i,3}=10000-1400*i;
end
for i=1:7
    links{
    
    i+7,1}=['浏览',num2str(i)];
    links{
    
    i+7,2}=['下载',num2str(i)];
    links{
    
    i+7,3}=900;
end
for i=1:7
    links{
    
    i+14,1}=['浏览',num2str(i)];
    links{
    
    i+14,2}=['流失',num2str(i)];
    links{
    
    i+14,3}=500;
    if i>=3
        links{
    
    i+14,3}=1100;
    end
end
for i=1:6
    links{
    
    i+21,1}=['下载',num2str(i)];
    links{
    
    i+21,2}=['浏览',num2str(i+2)];
    links{
    
    i+21,3}=600;
end
for i=1:6
    links{
    
    i+27,1}=['下载',num2str(i)];
    links{
    
    i+27,2}=['流失',num2str(i+1)];
    links{
    
    i+27,3}=300;
end

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

SK.NodeList={
    
    '浏览1','浏览2','浏览3','浏览4','浏览5','浏览6','浏览7','浏览8',...
             '下载1','下载2','下载3','下载4','下载5','下载6','下载7',...
             '流失1','流失2','流失3','流失4','流失5','流失6','流失7'};
SK.ColorList=[197,141,91;69,168,134;114,191,220;193,135,146;242,132,98;249,190,89;207,202,100;171,203,110;
              repmat([114,158,158],[7,1]);repmat([100,136,177],[7,1])]./255;

% 修改对齐方式(Set alignment)
% 'up'/'down'/'center'(default)
SK.Align='down';

% 修改链接颜色渲染方式(Set link color rendering method)
% 'left'/'right'/'interp'(default)/'map'/'simple'
SK.RenderingMethod='left'; 

% 修改文本位置(Set Text Location)
% 'left'(default)/'right'/'top'/'center'/'bottom'
SK.LabelLocation='right';

% 设置方块占比(Set the scale of blocks)
% BlockScale>0 & BlockScale<1
SK.BlockScale=.16;

% 开始绘图(Start drawing)
SK.draw()

% 循环设置标签属性(Loop Set Label Properties)
for i=1:22
    SK.setLabel(i,'FontName','宋体','FontSize',12)
end

for i=10:15
    SK.moveBlockY(i,(9-i).*1000);
end
for i=17:22
    SK.moveBlockY(i,(16-i).*1000);
end

Эффект трех методов выравнивания:

На самом деле, базовая версия без дополнительной настройки параметров тоже очень красива:

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

SK.NodeList={
    
    '浏览1','浏览2','浏览3','浏览4','浏览5','浏览6','浏览7','浏览8',...
             '下载1','下载2','下载3','下载4','下载5','下载6','下载7',...
             '流失1','流失2','流失3','流失4','流失5','流失6','流失7'};

% 修改对齐方式(Set alignment)
% 'up'/'down'/'center'(default)
SK.Align='top';

% 修改链接颜色渲染方式(Set link color rendering method)
% 'left'/'right'/'interp'(default)/'map'/'simple'
SK.RenderingMethod='left'; 

% 修改文本位置(Set Text Location)
% 'left'(default)/'right'/'top'/'center'/'bottom'
SK.LabelLocation='right';

% 开始绘图(Start drawing)
SK.draw()

% 循环设置标签属性(Loop Set Label Properties)
for i=1:22
    SK.setLabel(i,'FontName','宋体','FontSize',12)
end

12 Дополнительный малый кейс (2)

% 随机生成数据(Randomly generated data)
clc;clear;
SourceValue=randi([1,30],[1,9]);
LayerNum=[9,6,4,7,10];
links{
    
    1,3}='';
for k=1:4
    TargetValue=zeros(1,LayerNum(k+1));
    for i=1:LayerNum(k)
        tValue=randi([0,13],[1,LayerNum(k+1)]);
        tValue=tValue./sum(tValue).*SourceValue(i);
        for j=1:LayerNum(k+1)
            TargetValue(j)=TargetValue(j)+tValue(j);
            if tValue(j)>eps
                tLen=size(links,1);
                links{
    
    tLen+1,1}=[char(64+k),num2str(i)];
                links{
    
    tLen+1,2}=[char(64+k+1),num2str(j)];
                links{
    
    tLen+1,3}=tValue(j);
            end
        end
    end
    SourceValue=TargetValue;
end
links(1,:)=[];

% 创建桑基图对象(Create a Sankey diagram object)
SK=SSankey(links(:,1),links(:,2),links(:,3));

% 修改链接颜色渲染方式(Set link color rendering method)
% 'left'/'right'/'interp'(default)/'map'/'simple'
SK.RenderingMethod='interp';  

% 修改对齐方式(Set alignment)
% 'up'/'down'/'center'(default)
SK.Align='center';

% 修改文本位置(Set Text Location)
% 'left'(default)/'right'/'top'/'center'/'bottom'
SK.LabelLocation='top';

% 设置缝隙占比(Separation distance proportion)
SK.Sep=.4;

% 开始绘图(Start drawing)
SK.draw()

Вы можете настроить метод рендеринга самостоятельно:

Полный код функции инструмента

classdef SSankey < handle
% Copyright (c) 2023, Zhaoxu Liu / slandarer
% =========================================================================
% @author : slandarer
% 公众号  : slandarer随笔
% 知乎    : slandarer
% -------------------------------------------------------------------------
% Zhaoxu Liu / slandarer (2023). sankey plot 
% (https://www.mathworks.com/matlabcentral/fileexchange/128679-sankey-plot), 
% MATLAB Central File Exchange. 检索来源 2023/4/28.
    properties
        Source;Target;Value;
        SourceInd;TargetInd;
        Layer;LayerPos;
        AdjMat;BoolMat;
        RenderingMethod='interp'  % 'left'/'right'/'interp'/'map'/'simple'
        LabelLocation='left'      % 'left'/'right'/'top'/'center'/'bottom'
        Align='center'            % 'up'/'down'/'center'
        BlockScale=0.05;          %  BlockScale>0 ! !
        Sep=0.05;                 %  Sep>=0 ! !
        NodeList={
    
    };
        ColorList=[[65,140,240;252,180,65;224,64,10;5,100,146;191,191,191;26,59,105;255,227,130;18,156,221;
                    202,107,75;0,92,219;243,210,136;80,99,129;241,185,168;224,131,10;120,147,190]./255;
                   [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];
        BlockHdl;LinkHdl;LabelHdl;ax;Parent;
        BN;LN;VN;TotalLen;SepLen;
        arginList={
    
    'RenderingMethod','LabelLocation','BlockScale',...
                   'Sep','Align','ColorList','Parent','NameList'}
    end
% 构造函数 =================================================================
    methods
        function obj=SSankey(varargin)
            % 获取基本数据 -------------------------------------------------
            if isa(varargin{
    
    1},'matlab.graphics.axis.Axes')
                obj.ax=varargin{
    
    1};varargin(1)=[];
            else  
            end
            obj.Source=varargin{
    
    1};
            obj.Target=varargin{
    
    2};
            obj.Value=varargin{
    
    3};
            varargin(1:3)=[];
            % 获取其他信息 -------------------------------------------------
            for i=1:2:(length(varargin)-1)
                tid=ismember(obj.arginList,varargin{
    
    i});
                if any(tid)
                obj.(obj.arginList{
    
    tid})=varargin{
    
    i+1};
                end
            end
            if isempty(obj.ax)&&(~isempty(obj.Parent)),obj.ax=obj.Parent;end
            if isempty(obj.ax),obj.ax=gca;end
            obj.ax.NextPlot='add';
            % 基本数据预处理 -----------------------------------------------
            if isempty(obj.NodeList)
                obj.NodeList=[obj.Source;obj.Target];
                obj.NodeList=unique(obj.NodeList,'stable');
            end
            obj.BN=length(obj.NodeList);
            if length(obj.NodeList)>size(obj.ColorList,1)
                obj.ColorList=[obj.ColorList;rand(length(obj.NodeList),3).*.7];
            end
            obj.VN=length(obj.Value);
            % 坐标区域基础设置 ---------------------------------------------
            obj.ax.YDir='reverse';
            obj.ax.XColor='none';
            obj.ax.YColor='none';
        end
% Copyright (c) 2023, Zhaoxu Liu / slandarer
% =========================================================================
% @author : slandarer
% 公众号  : slandarer随笔
% 知乎    : slandarer
% -------------------------------------------------------------------------
% Zhaoxu Liu / slandarer (2023). sankey plot 
% (https://www.mathworks.com/matlabcentral/fileexchange/128679-sankey-plot), 
% MATLAB Central File Exchange. 检索来源 2023/4/28.
% 绘图函数 =================================================================
        function draw(obj)
            % 生成整体邻接矩阵 ---------------------------------------------
            obj.AdjMat=zeros(obj.BN,obj.BN);
            for i=1:length(obj.Source)
                obj.SourceInd(i)=find(strcmp(obj.Source{
    
    i},obj.NodeList));
                obj.TargetInd(i)=find(strcmp(obj.Target{
    
    i},obj.NodeList));
                obj.AdjMat(obj.SourceInd(i),obj.TargetInd(i))=obj.Value{
    
    i};
            end
            obj.BoolMat=abs(obj.AdjMat)>0;
            % 计算每个对象位于的层、每层方块长度、每个方块位置 ----------------
            obj.Layer=zeros(obj.BN,1);
            obj.Layer(sum(obj.BoolMat,1)==0)=1;
            startMat=diag(obj.Layer);
            for i=1:(obj.BN-1)
                tLayer=(sum(startMat*obj.BoolMat^i,1)>0).*(i+1);
                obj.Layer=max([obj.Layer,tLayer'],[],2);
            end
            obj.LN=max(obj.Layer);
            obj.TotalLen=max([sum(obj.AdjMat,1).',sum(obj.AdjMat,2)],[],2);
            obj.SepLen=max(obj.TotalLen).*obj.Sep;
            obj.LayerPos=zeros(obj.BN,4);
            for i=1:obj.LN
                tBlockInd=find(obj.Layer==i);
                tBlockLen=[0;cumsum(obj.TotalLen(tBlockInd))];
                tY1=tBlockLen(1:end-1)+(0:length(tBlockInd)-1).'.*obj.SepLen;
                tY2=tBlockLen(2:end)+(0:length(tBlockInd)-1).'.*obj.SepLen;
                obj.LayerPos(tBlockInd,3)=tY1;
                obj.LayerPos(tBlockInd,4)=tY2;
                % for j=1:length(tY2)
                %     plot([i,i],[tY1(j),tY2(j)],'LineWidth',2)
                % end
            end
            obj.LayerPos(:,1)=obj.Layer;
            obj.LayerPos(:,2)=obj.Layer+obj.BlockScale;
            % 根据对齐方式调整Y坐标 -----------------------------------------
            tMinY=min(obj.LayerPos(:,3));
            tMaxY=max(obj.LayerPos(:,4));
            for i=1:obj.LN
                tBlockInd=find(obj.Layer==i);
                tBlockPos3=obj.LayerPos(tBlockInd,3);
                tBlockPos4=obj.LayerPos(tBlockInd,4);
                switch obj.Align
                    case 'up'
                    case 'down'
                        obj.LayerPos(tBlockInd,3)=obj.LayerPos(tBlockInd,3)+tMaxY-max(tBlockPos4);
                        obj.LayerPos(tBlockInd,4)=obj.LayerPos(tBlockInd,4)+tMaxY-max(tBlockPos4);
                    case 'center'
                        obj.LayerPos(tBlockInd,3)=obj.LayerPos(tBlockInd,3)+...
                            min(tBlockPos3)/2-max(tBlockPos4)/2+tMinY/2-tMaxY/2;
                        obj.LayerPos(tBlockInd,4)=obj.LayerPos(tBlockInd,4)+...
                            min(tBlockPos3)/2-max(tBlockPos4)/2+tMinY/2-tMaxY/2;
                end
            end
            % 绘制连接 -----------------------------------------------------
            for i=1:obj.VN
                tSource=obj.SourceInd(i);
                tTarget=obj.TargetInd(i);
                tS1=sum(obj.AdjMat(tSource,1:(tTarget-1)))+obj.LayerPos(tSource,3);
                tS2=sum(obj.AdjMat(tSource,1:tTarget))+obj.LayerPos(tSource,3);
                tT1=sum(obj.AdjMat(1:(tSource-1),tTarget))+obj.LayerPos(tTarget,3);
                tT2=sum(obj.AdjMat(1:tSource,tTarget))+obj.LayerPos(tTarget,3);
                if isempty(tS1),tS1=0;end
                if isempty(tT1),tT1=0;end
                tX=[obj.LayerPos(tSource,1),obj.LayerPos(tSource,2),obj.LayerPos(tTarget,1),obj.LayerPos(tTarget,2)];
                qX=linspace(obj.LayerPos(tSource,1),obj.LayerPos(tTarget,2),200);qT=linspace(0,1,50);
                qY1=interp1(tX,[tS1,tS1,tT1,tT1],qX,'pchip');
                qY2=interp1(tX,[tS2,tS2,tT2,tT2],qX,'pchip');
                XX=repmat(qX,[50,1]);YY=qY1.*(qT'.*0+1)+(qY2-qY1).*(qT');
                MeshC=ones(50,200,3);
                switch obj.RenderingMethod
                    case 'left'
                        MeshC(:,:,1)=MeshC(:,:,1).*obj.ColorList(tSource,1);
                        MeshC(:,:,2)=MeshC(:,:,2).*obj.ColorList(tSource,2);
                        MeshC(:,:,3)=MeshC(:,:,3).*obj.ColorList(tSource,3);
                    case 'right'
                        MeshC(:,:,1)=MeshC(:,:,1).*obj.ColorList(tTarget,1);
                        MeshC(:,:,2)=MeshC(:,:,2).*obj.ColorList(tTarget,2);
                        MeshC(:,:,3)=MeshC(:,:,3).*obj.ColorList(tTarget,3);
                    case 'interp'
                        MeshC(:,:,1)=repmat(linspace(obj.ColorList(tSource,1),obj.ColorList(tTarget,1),200),[50,1]);
                        MeshC(:,:,2)=repmat(linspace(obj.ColorList(tSource,2),obj.ColorList(tTarget,2),200),[50,1]);
                        MeshC(:,:,3)=repmat(linspace(obj.ColorList(tSource,3),obj.ColorList(tTarget,3),200),[50,1]);
                    case 'map'
                        MeshC=MeshC(:,:,1).*obj.Value{
    
    i};
                    case 'simple'
                        MeshC(:,:,1)=MeshC(:,:,1).*.6;
                        MeshC(:,:,2)=MeshC(:,:,2).*.6;
                        MeshC(:,:,3)=MeshC(:,:,3).*.6;
                end
                obj.LinkHdl(i)=surf(obj.ax,XX,YY,XX.*0,'EdgeColor','none','FaceAlpha',.3,'CData',MeshC);
            end
            % 绘制方块 -----------------------------------------------------
            for i=1:obj.BN
                obj.BlockHdl(i)=fill(obj.ax,obj.LayerPos(i,[1,2,2,1]),...
                    obj.LayerPos(i,[3,3,4,4]),obj.ColorList(i,:),'EdgeColor','none');
            end
            % 绘制文本 -----------------------------------------------------
            for i=1:obj.BN
                switch obj.LabelLocation
                    case 'right'
                        obj.LabelHdl(i)=text(obj.ax,obj.LayerPos(i,2),mean(obj.LayerPos(i,[3,4])),...
                            [' ',obj.NodeList{
    
    i}],'FontSize',15,'FontName','Times New Roman','HorizontalAlignment','left');
                    case 'left'
                        obj.LabelHdl(i)=text(obj.ax,obj.LayerPos(i,1),mean(obj.LayerPos(i,[3,4])),...
                            [obj.NodeList{
    
    i},' '],'FontSize',15,'FontName','Times New Roman','HorizontalAlignment','right');
                    case 'top'
                        obj.LabelHdl(i)=text(obj.ax,mean(obj.LayerPos(i,[1,2])),obj.LayerPos(i,3),...
                            obj.NodeList{
    
    i},'FontSize',15,'FontName','Times New Roman','HorizontalAlignment','center','VerticalAlignment','bottom');
                    case 'center'
                        obj.LabelHdl(i)=text(obj.ax,mean(obj.LayerPos(i,[1,2])),mean(obj.LayerPos(i,[3,4])),...
                            obj.NodeList{
    
    i},'FontSize',15,'FontName','Times New Roman','HorizontalAlignment','center');
                    case 'bottom'
                        obj.LabelHdl(i)=text(obj.ax,mean(obj.LayerPos(i,[1,2])),obj.LayerPos(i,4),...
                            obj.NodeList{
    
    i},'FontSize',15,'FontName','Times New Roman','HorizontalAlignment','center','VerticalAlignment','top');
                end
            end
            % -------------------------------------------------------------
            axis tight;help SSankey
        end
% Copyright (c) 2023, Zhaoxu Liu / slandarer
% =========================================================================
% @author : slandarer
% 公众号  : slandarer随笔
% 知乎    : slandarer
% -------------------------------------------------------------------------
% Zhaoxu Liu / slandarer (2023). sankey plot 
% (https://www.mathworks.com/matlabcentral/fileexchange/128679-sankey-plot), 
% MATLAB Central File Exchange. 检索来源 2023/4/28.
% =========================================================================
        function setBlock(obj,n,varargin)
            set(obj.BlockHdl(n),varargin{
    
    :})
        end
        function setLink(obj,n,varargin)
            set(obj.LinkHdl(n),varargin{
    
    :})
        end
        function setLabel(obj,n,varargin)
            set(obj.LabelHdl(n),varargin{
    
    :})
        end
        function moveBlockY(obj,n,dy)
            obj.LayerPos(n,[3,4])=obj.LayerPos(n,[3,4])-dy;
            set(obj.BlockHdl(n),'YData',obj.LayerPos(n,[3,3,4,4]));
            switch obj.LabelLocation
                case 'right',set(obj.LabelHdl(n),'Position',[obj.LayerPos(n,2),mean(obj.LayerPos(n,[3,4]))]);
                case 'left',set(obj.LabelHdl(n),'Position',[obj.LayerPos(n,1),mean(obj.LayerPos(n,[3,4]))]);
                case 'top',set(obj.LabelHdl(n),'Position',[mean(obj.LayerPos(n,[1,2])),obj.LayerPos(n,3)]);
                case 'center',set(obj.LabelHdl(n),'Position',[mean(obj.LayerPos(n,[1,2])),mean(obj.LayerPos(n,[3,4]))]);
                case 'bottom',set(obj.LabelHdl(n),'Position',[mean(obj.LayerPos(n,[1,2])),obj.LayerPos(n,4)]);
            end
            for i=1:obj.VN
                tSource=obj.SourceInd(i);
                tTarget=obj.TargetInd(i);
                if tSource==n||tTarget==n
                    tS1=sum(obj.AdjMat(tSource,1:(tTarget-1)))+obj.LayerPos(tSource,3);
                    tS2=sum(obj.AdjMat(tSource,1:tTarget))+obj.LayerPos(tSource,3);
                    tT1=sum(obj.AdjMat(1:(tSource-1),tTarget))+obj.LayerPos(tTarget,3);
                    tT2=sum(obj.AdjMat(1:tSource,tTarget))+obj.LayerPos(tTarget,3);
                    if isempty(tS1),tS1=0;end
                    if isempty(tT1),tT1=0;end
                    tX=[obj.LayerPos(tSource,1),obj.LayerPos(tSource,2),obj.LayerPos(tTarget,1),obj.LayerPos(tTarget,2)];
                    qX=linspace(obj.LayerPos(tSource,1),obj.LayerPos(tTarget,2),200);qT=linspace(0,1,50);
                    qY1=interp1(tX,[tS1,tS1,tT1,tT1],qX,'pchip');
                    qY2=interp1(tX,[tS2,tS2,tT2,tT2],qX,'pchip');
                    YY=qY1.*(qT'.*0+1)+(qY2-qY1).*(qT');
                    set(obj.LinkHdl(i),'YData',YY);
                end
            end
        end
    end
% Copyright (c) 2023, Zhaoxu Liu / slandarer
% =========================================================================
% @author : slandarer
% 公众号  : slandarer随笔
% 知乎    : slandarer
% -------------------------------------------------------------------------
% Zhaoxu Liu / slandarer (2023). sankey plot 
% (https://www.mathworks.com/matlabcentral/fileexchange/128679-sankey-plot), 
% MATLAB Central File Exchange. 检索来源 2023/4/28.
end

над

Написать этот код и выбрать кейс на самом деле непросто, надеюсь, он всем понравится 点赞и будет просмотрен! !

Пожалуйста, не используйте этот код в коммерческих целях без разрешения.Если вы его цитируете, вы можете указать ссылку на мой файлообменник.Можно использовать следующий формат:

Zhaoxu Liu / slandarer (2023).Sankey plot ( https://www.mathworks.com/matlabcentral/fileexchange/128679-sankey-plot ), MATLAB Central File Exchange. Получено от 2023/4/28.

Если вы будете перепечатывать, пожалуйста, сохраните вышеуказанную ссылку для обмена файлами и ссылку на эту статью! ! ! ! !

Инструмент можно получить по ссылке выше для обмена файлами или загрузить через следующий репозиторий gitee:

https://gitee.com/slandarer/matlab-sankey-диаграмма

Supongo que te gusta

Origin blog.csdn.net/slandarer/article/details/130430147
Recomendado
Clasificación
Diario
Más
2024-05-25(0)
2024-05-24(13)
2024-05-23(34)
2024-05-22(10)
2024-05-21(34)
2024-05-20(5)
2024-05-19(0)
2024-05-18(30)
2024-05-17(4)
2024-05-16(22)

Code World

© 2018 codetd.com