注意
转到末尾下载完整示例代码。
Sankey 类#
通过生成三个基本图表来演示 Sankey 类。
import matplotlib.pyplot as plt
from matplotlib.sankey import Sankey
示例 1 -- 大部分使用默认设置
这演示了如何通过隐式调用 Sankey.add() 方法并将 finish() 附加到类的调用来创建简单的图表。

注意
实例化 Sankey() 时未提供 Axes,因此它们是自动创建的。
由于数据已归一化,因此不需要 scale 参数。
默认情况下,路径的长度是对齐的。
示例 2
这演示了
设置一条路径比其他路径长
在图表中间放置标签
使用 scale 参数归一化流量
隐式地将关键字参数传递给 PathPatch()
改变箭头头的角度
改变路径尖端与其标签之间的偏移
格式化路径标签中的数字和相关单位
在图形创建后改变补丁和标签的外观
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, xticks=[], yticks=[],
title="Flow Diagram of a Widget")
sankey = Sankey(ax=ax, scale=0.01, offset=0.2, head_angle=180,
format='%.0f', unit='%')
sankey.add(flows=[25, 0, 60, -10, -20, -5, -15, -10, -40],
labels=['', '', '', 'First', 'Second', 'Third', 'Fourth',
'Fifth', 'Hurray!'],
orientations=[-1, 1, 0, 1, 1, 1, -1, -1, 0],
pathlengths=[0.25, 0.25, 0.25, 0.25, 0.25, 0.6, 0.25, 0.25,
0.25],
patchlabel="Widget\nA") # Arguments to matplotlib.patches.PathPatch
diagrams = sankey.finish()
diagrams[0].texts[-1].set_color('r')
diagrams[0].text.set_fontweight('bold')

注意
由于流量之和不为零,因此主干的宽度不均匀。matplotlib 日志系统会在 DEBUG 级别记录此信息。
第二个流量没有显示,因为它的值为零。同样,这会在 DEBUG 级别记录。
示例 3
这演示了
连接两个系统
关闭数量标签
添加图例
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, xticks=[], yticks=[], title="Two Systems")
flows = [0.25, 0.15, 0.60, -0.10, -0.05, -0.25, -0.15, -0.10, -0.35]
sankey = Sankey(ax=ax, unit=None)
sankey.add(flows=flows, label='one',
orientations=[-1, 1, 0, 1, 1, 1, -1, -1, 0])
sankey.add(flows=[-0.25, 0.15, 0.1], label='two',
orientations=[-1, -1, -1], prior=0, connect=(0, 0))
diagrams = sankey.finish()
diagrams[-1].patch.set_hatch('/')
plt.legend()

请注意,只指定了一个连接,但系统形成了一个回路,因为:(1) 路径的长度是对齐的,(2) 流量的方向和顺序是镜像的。
plt.show()
参考
本示例展示了以下函数、方法、类和模块的使用
脚本总运行时间: (0 分钟 1.537 秒)