要创建一个桑基图(Sankey diagram)可以使用Plotly库来实现。以下是一个简单的示例代码,展示如何使用Plotly创建一个基本的桑基图:
import plotly.graph_objects as go
fig = go.Figure(data=[go.Sankey(
node=dict(
pad=15,
thickness=20,
line=dict(color="black", width=0.5),
label=["A", "B", "C", "D", "E"],
color="blue"
),
link=dict(
source=[0, 1, 1, 2, 3],
target=[1, 2, 3, 4, 4],
value=[8, 4, 2, 8, 4]
)
)])
fig.show()
在这个示例中,我们首先导入了plotly.graph_objects库,并创建了一个Figure对象。然后,我们使用go.Sankey()函数创建了一个Sankey对象,并传入了节点和链接的相关信息。节点信息包括标签和颜色,链接信息包括源节点、目标节点和值。最后,我们使用fig.show()方法显示出这个Sankey图。
通过调整节点和链接的信息以及样式设置,可以创建出不同样式和结构的Sankey图。Plotly提供了丰富的配置选项,可以根据需要自定义Sankey图的外观和布局。