要使用Plotly的Sankey类,需要首先安装Plotly库。你可以使用以下命令安装Plotly库:
pip install plotly
接下来,可以使用以下代码示例创建一个Sankey图:
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, 2, 2, 3],
target=[2, 3, 4, 5, 4],
value=[8, 4, 2, 8, 4]
)
)])
fig.show()
上面的代码创建一个简单的Sankey图,其中包含5个节点和5条连接线。您可以根据需要调整节点和连接线的属性来定制Sankey图。最后,使用fig.show()
方法显示Sankey图。