python 桑基图

桑基图（Sankey diagram）是一种可视化工具，用于表示流量、能量或数量的流动。在Python中，可以使用多种库来绘制桑基图，其中比较常用的是`matplotlib`和`plotly`。

使用`matplotlib`绘制桑基图的步骤如下：

1. 安装`matplotlib`库：可以使用`pip install matplotlib`命令进行安装。

2. 导入所需的库和模块：

import matplotlib.pyplot as plt
from matplotlib.sankey import Sankey

3. 创建数据：定义一个包含节点和流量的列表。

nodes = ['Node A', 'Node B', 'Node C', 'Node D']
flows = [20, 15, 10, 5]

4. 创建画布和坐标轴：

fig, ax = plt.subplots()

5. 绘制桑基图：

sankey = Sankey(ax=ax, scale=0.5, offset=0.3, head_angle=180)
sankey.add(flows=[20, -15, -5, -10], labels=nodes, orientations=[0, 1, 1, 0])
sankey.finish()

6. 显示图形：

plt.show()

这样就可以绘制一个简单的桑基图。你可以根据自己的需求添加更多的节点和流量，调整样式和颜色等。

另外，如果你想使用`plotly`库绘制桑基图，可以参考官方文档以了解更多信息。