注意

转到结尾 下载完整的示例代码。

Sankey 类#

通过生成三个基本图表来演示 Sankey 类。

import matplotlib.pyplot as plt

from matplotlib.sankey import Sankey

示例 1 -- 主要使用默认值

演示如何通过隐式调用 Sankey.add() 方法并通过在对类的调用中附加 finish() 来创建简单图表。

Sankey(flows=[0.25, 0.15, 0.60, -0.20, -0.15, -0.05, -0.50, -0.10],
       labels=['', '', '', 'First', 'Second', 'Third', 'Fourth', 'Fifth'],
       orientations=[-1, 1, 0, 1, 1, 1, 0, -1]).finish()
plt.title("The default settings produce a diagram like this.")
The default settings produce a diagram like this.

注意

  1. 在实例化 Sankey() 时没有提供坐标轴，因此它们是自动创建的。

  2. 由于数据已经归一化，因此 scale 参数不是必需的。

  3. 默认情况下，路径的长度是经过调整的。

示例 2

演示了以下内容：

  1. 将一条路径设置为比其他路径更长

  2. 将标签放置在图表中间

  3. 使用 scale 参数来归一化流量

  4. 隐式地将关键字参数传递给 PathPatch()

  5. 更改箭头头的角度

  6. 更改路径尖端与其标签之间的偏移量

  7. 格式化路径标签中的数字和关联的单位

  8. 在创建图形后更改补丁和标签的外观

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')
Flow Diagram of a Widget

注意

  1. 由于流量总和不为零，因此主干的宽度不均匀。matplotlib 日志记录系统在 DEBUG 级别的日志中记录了这一点。

  2. 第二条流量没有出现，因为其值为零。同样，这在 DEBUG 级别的日志中记录了。

示例 3

演示了以下内容：

  1. 连接两个系统

  2. 关闭数量标签

  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()
Two Systems

请注意，只指定了一个连接，但系统形成一个回路，因为： (1) 路径的长度经过调整， (2) 流量方向和顺序是镜像的。

plt.show()

参考资料

本示例中展示了以下函数、方法、类和模块的使用。

由 Sphinx-Gallery 生成的图库