python数据可视化技巧的100个练习 -- 23. 创建能量流的桑基图

2026-10-11 06:17:55

重要性★★★★☆

难度★★★☆☆

你是一家能源公司的数据分析师，该公司希望可视化来自不同能源（如煤炭、天然气和太阳能）到各个部门（如住宅、商业和工业）的能量流动。

你的任务是创建一个桑基图，以显示从不同能源到各个部门的能量流动。

数据应在代码中生成。

这个练习的目的是练习 Python 数据操作和可视化。

请编写代码，使用 Python 创建桑基图。

【数据生成代码示例】

import pandas as pdimport plotly.graph_objects as godata = {    'source': ['Coal', 'Coal', 'Natural Gas', 'Natural Gas', 'Solar'],    'target': ['Residential', 'Commercial', 'Residential', 'Industrial', 'Commercial'],    'value': [100, 200, 150, 250, 300]}df = pd.DataFrame(data)

【图表答案】

【代码答案】

import pandas as pdimport plotly.graph_objects as godata = {    'source': ['Coal', 'Coal', 'Natural Gas', 'Natural Gas', 'Solar'],    'target': ['Residential', 'Commercial', 'Residential', 'Industrial', 'Commercial'],    'value': [100, 200, 150, 250, 300]}df = pd.DataFrame(data)labels = list(set(df['source']).union(set(df['target'])))label_to_index = lambda x: labels.index(x)source_indices = [label_to_index(x) for x in df['source']]target_indices = [label_to_index(x) for x in df['target']]fig = go.Figure(data=[go.Sankey(    node=dict(        pad=15,        thickness=20,        line=dict(color="black", width=0.5),        label=labels    ),    link=dict(        source=source_indices,        target=target_indices,        value=df['value']    ))])fig.update_layout(title_text="能量流桑基图", font_size=10)fig.show()