简体   繁体   中英

How to split y axis labels and color each part separately?

ax.yaxis.get_major_ticks() allows me to colour-code each label differently. But I'm failing to split the label and mark each part with a different colour.

Example Image: A_B => A in Blue and _B in Red, similarly C_D => C in Blue and D in Red etc.

While looping through all the ticks the text is available with get_text() , but color coding each part separately is not possible with the same.

This a sample graphical representation of a horizontally stacked bar chart:

图片

Borrowing some code from this excellent post , text can be put together via offset boxes.

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

结果图

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM