python - 在 matplotlib 中连接两个桑基图

Question

我正在尝试使用 matplotlib 表示一个国家的天然气平衡。

这个想法是，我想使用一个 Sankey 绘制三个进口天然气来源，并将其连接到另一个 Sankey，该 Sankey 具有其他天然气来源(生产、天然气储存)和天然气消费者作为流出。

我尝试了很多次，但是我无法将两个图表连接在一起。

每个图表均按设计单独绘制。但是，一旦我添加 "prior=0, connect=(3,0)" ，据称它将两个图连接在一起，一切都会出错，给我带来了我无法完全理解的错误。这是代码。

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


ImportLabels=["Imports by\nNaftogaz","Imports by\nOstchem","Imports from\nEurope", ""]
ImportFlows=[11.493,9.768,1.935,-23.196]
ImportOrientation=[0,1,-1,0]

l=["Imports","Private\nextraction","State\nextraction","Net supplies\ntoUGS","Households","TKE","Metallurgy","Energy","Other\nIndustries","Technological\nlosses"]
v=[23.196,3.968,13.998,-2.252,-13.289,-7.163,-3.487,-4.72,-7.037,-3.17]
d=[0,-1,1,-1,0,0,1,1,1,-1]

sankey=Sankey(scale=1.0/69,patchlabel="Gas balance",format='%.1f',margin=0.15)
sankey.add(flows=ImportFlows, labels=ImportLabels, orientations=ImportOrientation, label='Imports',fc='#00AF00')
sankey.add(flows=v,labels=l,orientations=d, prior=0, connect=(3,0),label='Second',fc='#008000')

这个想法是将第一个图(其值为 -23.196)的 3 个流出与第二个 sankey(也有 23.196)的 0 个流入连接起来

这是错误文本:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-24-1220fede42ce> in <module>()
     14 sankey=Sankey(scale=1.0/69,patchlabel="Gas balance",format='%.1f',margin=0.15)
     15 sankey.add(flows=ImportFlows, labels=ImportLabels,  orientations=ImportOrientation, label='Imports',fc='#00AF00')
---> 16 sankey.add(flows=v,labels=l,orientations=d, prior=0, connect=(3,0),label='Second',fc='#008000')

C:\Python27\lib\site-packages\matplotlib\sankey.pyc in add(self, patchlabel, flows, orientations, labels, trunklength, pathlengths, prior, connect, rotation, **kwargs)
    369                    ("The connection index to the source diagram is %d, but "
    370                     "that diagram has only %d flows.\nThe index is zero-based."
--> 371                     % connect[0], len(self.diagrams[prior].flows))
    372             assert connect[1] < n, ("The connection index to this diagram is "
    373                                     "%d, but this diagram has only %d flows.\n"

TypeError: not enough arguments for format string

所以我不确定两个图之间的连接是否有问题(由 sankey.pyc 试图显示的文本建议)，或者 matplotlib 本身是否有问题，如 "TypeError: not enough argument for格式字符串”表示？

Answer 1

您的问题是您进行了两次 .add() 调用。第一次调用 Sankey() 已经构建了一个图表(默认灰色，有 1 个流入和 1 个流出)。因此，当您尝试连接到第一个图时，它会失败，因为它只有一个流，而您正在尝试连接到第三个流。 (无论如何它都会失败，因为流程不匹配。)

您需要在第一次调用中设置第一个图表，并且只有一个添加调用，例如:

sankey = Sankey(scale=1.0/69,patchlabel="Gas balance",format='%.1f',margin=0.15,
                flows=ImportFlows, labels=ImportLabels, 
                orientations=ImportOrientation, label='Imports',fc='#00AF00')
sankey.add(flows=v,labels=l,orientations=d, label='Second',fc='#008000', prior=0,
           connect=(3, 0))

这给了我: