使用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()