如何拆分y轴标签并分别为每个部分着色?

Kun*_*Roy 4 python matplotlib

ax.yaxis.get_major_ticks()允许我对每个标签进行不同的颜色编码。但是我没有拆分标签并用不同的颜色标记每个部分。

示例图像:A_B => A为蓝色,_B为红色,类似的C_D => C为蓝色,D为红色等。

在循环遍历所有刻度时,文本可用get_text(),但单独对每个部分进行颜色编码是不可能的。

这是水平堆叠条形图的示例图形表示:

图片

Joh*_*anC 6

这篇优秀的文章中借用一些代码,文本可以通过偏移框放在一起。

import matplotlib.pyplot as plt
from matplotlib.offsetbox import AnchoredOffsetbox, TextArea, HPacker

fig, ax = plt.subplots()

vals1 = [0.3, 0.5, 0.4, 0.2, 0.5]
vals2 = [0.2, 0.3, 0.2, 0.2, 0.1]
labels1 = ['A', 'B', 'CCCC', 'DDDDDD', 'E']
labels2 = ['B', 'CCCC', 'DDDDDD', 'E', 'F']
color1 = 'dodgerblue'
color2 = 'crimson'

ax.barh(range(len(vals1)), vals1, color=color1)
ax.barh(range(len(vals2)), vals2, left=vals1, color=color2)

ax.set_yticklabels([])
for i in range(len(labels1)):
    boxes = [TextArea(text, textprops=dict(color=color))
             for text, color in zip([labels1[i], '_', labels2[i]], [color1, 'black', color2])]
    xbox = HPacker(children=boxes, align="right", pad=1, sep=1)
    anchored_xbox = AnchoredOffsetbox(loc='center right', child=xbox, pad=0, frameon=False, bbox_to_anchor=(0, i),
                                      bbox_transform=ax.transData, borderpad=1)
    ax.add_artist(anchored_xbox)

plt.tight_layout()
plt.show()
Run Code Online (Sandbox Code Playgroud)

结果图