编辑 Python Matplotlib Sankey（删除值、对齐文本..）

Question

我是 Matplotlib 中 Sankey 图函数的新手，我希望有人能够帮助我解决一些问题，因为我似乎无法弄清楚。

首先：我用下面的例子来了解如何创建桑基图： https://flothesof.github.io/sankey-tutorial-matplotlib.html

这是桑基图：

现在我需要更改一些内容，其中两个我不知道如何更改。

我想删除图表中间显示值的标签（因此 14460、9720、7047、3059 和 2149）。
中间的所有文字都是粗体，除了最后一个（下面黄色部分）。我怎样才能把这个也加粗？

希望有人能提供帮助。

Answer 1

每个图表的 .text 字段包含其中央标签 Text object。 .texts 是每个 entering/leaving 箭头的 Text 对象的列表。您可以检查字符串以了解它是关于哪个文本的，并更改 bold 或 xy-position.

等属性

由于 Sankey 图非常复杂，因此 fine-tuning 创建图表后可能会出现位置。一种简单的方法是在字符串中添加额外的换行符。

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

fig, ax = plt.subplots(figsize=(8, 12))
ax.set_xticks([])
ax.set_yticks([])
ax.set_title("My funnel")
ax.set_axis_off()
number = [round(12345 * 0.6 ** i) for i in range(6)]
labels = ["\n\n\n\n\n\n\nTotal joined", "\n\n\n\nFirst", "\n\nSecond", "\nThird", "\nFourth", "\n\nReached final"]

sankey = Sankey(ax=ax, scale=0.0015, offset=0.3)
for input_number, output_number, label in zip(number[:-1], number[1:], labels):
    this_index = len(sankey.diagrams)
    prior = this_index - 1 if this_index > 0 else None
    pathlengths = [0, 0, 2 if this_index != 4 else 10]
    exitlabel = labels[-1] if this_index == 4 else None
    sankey.add(flows=[input_number, -output_number, output_number - input_number],
               orientations=[0, 0, 1],
               patchlabel=label,
               labels=['', exitlabel, 'quit'],
               prior=prior,
               connect=(1, 0),
               pathlengths=pathlengths,
               trunklength=10.,
               rotation=-90,
               facecolor=plt.cm.spring((this_index + 1) / 5))
diagrams = sankey.finish()
for diagram in diagrams:
    diagram.text.set_fontweight('bold')
    diagram.text.set_fontsize('10')
    for text in diagram.texts:
        text.set_fontsize('10')
        str = text.get_text()  # can be normal label, "quit", a number or empty
        if len(str) > 0:
            if str[0].isdigit():
                text.set_visible(False)
            elif str.startswith('quit'):
                xy = text.get_position()
                text.set_position((xy[0] + 2, xy[1]))
            else:
                text.set_fontweight('bold')
plt.ylim(ymin=plt.ylim()[0] * 1.05)
plt.show()

编辑 Python Matplotlib Sankey（删除值、对齐文本..）

Editing Python Matplotlib Sankey (removing values, align texts..)

matplotlib

python-3.x

sankey-diagram