将节点计数添加到Plotly Sankey图

import plotly.graph_objects as go

# 定义节点
nodes = [
    go.Sankey.Node(
        label="Node A",  # 节点标签
        value=10,       # 节点计数
        color="blue"    # 节点颜色
    ),
    go.Sankey.Node(
        label="Node B",
        value=5,
        color="red"
    ),
    go.Sankey.Node(
        label="Node C",
        value=8,
        color="green"
    )
]

# 定义连接
links = [
    go.Sankey.Link(
        source=0,   # 连接起点索引
        target=1,   # 连接终点索引
        value=5     # 连接计数
    ),
    go.Sankey.Link(
        source=1,
        target=2,
        value=3
    )
]

# 创建Sankey图对象
fig = go.Figure(data=[go.Sankey(node=dict(
    pad=15,   # 节点间距
    thickness=20,   # 节点厚度
    line=dict(color="black", width=0.5),   # 连接线的颜色和宽度
    label=["Node A", "Node B", "Node C"],  # 节点标签
    color=["blue", "red", "green"]  # 节点颜色
    ),
    link=dict(source=[0, 1],  # 连接起点索引
    target=[1, 2],   # 连接终点索引
    value=[5, 3]  # 连接计数
    ))])

# 显示Sankey图
fig.show()