Python编程小技巧--桑基图(Sankey Diagram)

# -*- coding: utf-8 -*-import plotly.graph_objects as goimport webbrowserimport osfrom collections import defaultdict# ================= 数据准备 =================labels = ["首页", "产品页", "购物车", "支付页", "完成", "离开"]colors = ["#1f77b4", "#ff7f0e", "#2ca02c", "#d62728", "#9467bd", "#8c564b"]source = [0, 0, 1, 1, 2, 2, 3]target = [1, 5, 2, 5, 3, 5, 4]values = [1000, 200, 800, 100, 600, 50, 550]# ================= 转化率计算 =================# 1. 计算每个源节点的总流出量outflow_sum = defaultdict(int)for s, v in zip(source, values):    outflow_sum[s] += v# 2. 计算每个连接的转化率 = 当前值 / 源节点总流出量conversion_rates = []for s, v in zip(source, values):    rate = v / outflow_sum[s] * 100    conversion_rates.append(rate)# 3. 生成连线标签：显示人数 + 转化率link_labels = [f"{v}人\n{rate:.1f}%" for v, rate in zip(values, conversion_rates)]# 4. 计算每个节点的总流入量（用于节点标签）inflow_sum = defaultdict(int)for t, v in zip(target, values):    inflow_sum[t] += vnode_labels_with_count = []for i, label in enumerate(labels):    inflow = inflow_sum.get(i, 0)    outflow = outflow_sum.get(i, 0)    if inflow > 0 and outflow > 0:        node_labels_with_count.append(f"{label}<br>流入:{inflow} 流出:{outflow}")    elif inflow > 0:        node_labels_with_count.append(f"{label}<br>流入:{inflow}")    elif outflow > 0:        node_labels_with_count.append(f"{label}<br>流出:{outflow}")    else:        node_labels_with_count.append(label)# ================= 创建桑基图 =================fig = go.Figure(data=[go.Sankey(    node=dict(        pad=25,        thickness=35,        line=dict(color="black", width=1),        label=node_labels_with_count,  # 节点显示流入流出量        color=colors,        hovertemplate="节点：%{label}<br> 流量：%{value}人<extra></extra>"    ),    link=dict(        source=source,        target=target,        value=values,        # 连线颜色：根据转化率设置深浅（转化率高=颜色深）        color=[f"rgba(31, 119, 180, {0.2 + rate/200})" for rate in conversion_rates],        label=link_labels,  # 显示人数+转化率        hovertemplate=(            "路径：%{source.label} → %{target.label}<br>"            "流量：%{value}人<br>"            "转化率：%{customdata:.1f}%<extra></extra>"        ),        customdata=conversion_rates  # 传递转化率数据用于悬停显示    ))])# ================= 布局优化 =================fig.update_layout(    title=dict(        text="用户浏览路径分析（含转化率）",        font_size=20,        font_color="#333333",        x=0.5,        xanchor="center",        y=0.95,        yanchor="top"    ),    font=dict(size=13, family="Microsoft YaHei, SimHei, sans-serif"),    height=750,    width=1200,    plot_bgcolor="white",    paper_bgcolor="white",    margin=dict(l=30, r=30, t=90, b=30))# ================= 添加图例说明 =================fig.add_annotation(    text="提示：连线标签显示「人数 + 转化率」，悬停查看详细信息",    xref="paper", yref="paper",    x=0.5, y=-0.08,    showarrow=False,    font=dict(size=11, color="#666666"))# 添加转化率图例（颜色深浅示意）fig.add_annotation(    text="颜色深浅 = 转化率高低",    xref="paper", yref="paper",    x=0.98, y=0.02,    showarrow=False,    font=dict(size=10, color="#999999"),    xanchor="right")# ================= 保存并打开 =================output_file = os.path.abspath("sankey_with_conversion.html")fig.write_html(output_file)