python 桑基图实例

以下是一个 Python 桑基图实例：

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

# 设置桑基图参数
flows = [0.2, 0.1, 0.3, -0.1, -0.2, -0.1]
labels = ['A', 'B', 'C', 'D', 'E', 'F']

# 绘制桑基图
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, xticks=[], yticks=[], title="Sankey Diagram")
sankey = Sankey(ax=ax, unit=None, offset=0.3, head_angle=180,
                format='%.1f', gap=0.2, scale=1.0, shoulder=0.0)
sankey.add(flows=flows, labels=labels)

# 显示图像
diagrams = sankey.finish()
plt.show()

该代码将绘制一个包含 6 个节点的桑基图，其中前三个节点代表流入的物质，后三个节点代表流出的物质。在图中，A、B、C、D、E 和 F 分别代表 6 个节点。flows 列表中的正数代表流入的物质量，负数代表流出的物质量。在本例中，A、B、C 为流入节点，D、E、F 为流出节点。流入节点的物质总量为 0.6，流出节点的物质总量为 0.4。