使用Sankey的长链连接

通过建立长链连接来演示/测试Sankey类。

  1. import matplotlib.pyplot as plt
  2. from matplotlib.sankey import Sankey
  4. links_per_side = 6
  7. def side(sankey, n=1):
  8.     """Generate a side chain."""
  9.     prior = len(sankey.diagrams)
  10.     for i in range(0, 2*n, 2):
  11.         sankey.add(flows=[1, -1], orientations=[-1, -1],
  12.                    patchlabel=str(prior + i),
  13.                    prior=prior + i - 1, connect=(1, 0), alpha=0.5)
  14.         sankey.add(flows=[1, -1], orientations=[1, 1],
  15.                    patchlabel=str(prior + i + 1),
  16.                    prior=prior + i, connect=(1, 0), alpha=0.5)
  19. def corner(sankey):
  20.     """Generate a corner link."""
  21.     prior = len(sankey.diagrams)
  22.     sankey.add(flows=[1, -1], orientations=[0, 1],
  23.                patchlabel=str(prior), facecolor='k',
  24.                prior=prior - 1, connect=(1, 0), alpha=0.5)
  27. fig = plt.figure()
  28. ax = fig.add_subplot(1, 1, 1, xticks=[], yticks=[],
  29.                      title="Why would you want to do this?\n(But you could.)")
  30. sankey = Sankey(ax=ax, unit=None)
  31. sankey.add(flows=[1, -1], orientations=[0, 1],
  32.            patchlabel="0", facecolor='k',
  33.            rotation=45)
  34. side(sankey, n=links_per_side)
  35. corner(sankey)
  36. side(sankey, n=links_per_side)
  37. corner(sankey)
  38. side(sankey, n=links_per_side)
  39. corner(sankey)
  40. side(sankey, n=links_per_side)
  41. sankey.finish()
  42. # Notice:
  43. # 1. The alignment doesn't drift significantly (if at all; with 16007
  44. #    subdiagrams there is still closure).
  45. # 2. The first diagram is rotated 45 deg, so all other diagrams are rotated
  46. #    accordingly.
  48. plt.show()

使用Sankey的长链连接示例

参考

此示例中显示了以下函数，方法，类和模块的使用：

  1. import matplotlib
  2. matplotlib.sankey
  3. matplotlib.sankey.Sankey
  4. matplotlib.sankey.Sankey.add
  5. matplotlib.sankey.Sankey.finish

下载这个示例