使用python绘制桑基图

本文主要是介绍使用python绘制桑基图，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

使用python绘制桑基图

桑基图
效果
代码

桑基图

桑基图（Sankey Diagram）是一种用来表示流动（如能源、资金、材料等）在不同实体之间转移的图表。 每个流的宽度与流量成正比，通常用于显示能量或成本流动的分布情况。

桑基图的特征：

节点 (Nodes)：表示实体或状态。
链接 (Links)：表示从一个节点流向另一个节点的流动，宽度表示流量的大小。

效果

在这里插入图片描述

代码

import plotly.graph_objects as go# 定义节点标签
labels = ["煤炭", "天然气", "核能", "太阳能", "风能", "居民用电", "商业用电", "工业用电", "交通运输"]# 定义链接：源节点，目标节点，流量
sources = [0, 1, 2, 3, 4, 0, 1, 1, 2, 3, 4, 0, 1, 2, 3, 4]
targets = [5, 5, 5, 5, 5, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 8]
values = [10, 15, 5, 10, 7, 8, 10, 5, 12, 5, 3, 10, 8, 7, 3, 2]# 定义链接颜色
link_colors = ["#FF5733", "#33FF57", "#3357FF", "#F333FF", "#FF33A6","#FFBD33", "#33FFBD", "#FF5733", "#57FF33", "#5733FF","#F333FF", "#FF5733", "#33FF57", "#3357FF", "#F333FF", "#FF33A6"]# 创建桑基图
fig = go.Figure(go.Sankey(node=dict(pad=15,  # 节点之间的填充距离thickness=20,  # 节点的厚度line=dict(color="black", width=0.5),  # 节点的边界颜色和宽度label=labels,  # 节点标签),link=dict(source=sources,  # 源节点target=targets,  # 目标节点value=values,  # 流量值color=link_colors  # 链接颜色)
))# 更新布局
fig.update_layout(title_text="能源流动桑基图", font_size=10)# 显示图表
fig.show()