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