Выровнять метки значений на горизонтальной гистограмме с накоплением (Matplotlib)

Question

Используя следующий код, я добавил метки значений к горизонтальной гистограмме в Matplotlib:

import pandas
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

def sumzip(*items):
    return [sum(values) for values in zip(*items)]

fig, ax = plt.subplots(figsize=(10,6))

N = 5
values1 = [130, 120, 170, 164, 155]
values2 = [120, 185, 162, 150, 153]
values3 = [100, 170, 160, 145, 150]

ind = np.arange(N) + .15
width = 0.3

rects1 = plt.barh(ind, values1, width, color='blue') 
rects2 = plt.barh(ind, values2, width, left = sumzip(values1), color='green') 
rects3 = plt.barh(ind, values3, width, left = sumzip(values1, values2), color='red')

extra_space = 0.15
ax.set_yticks(ind+width-extra_space)
ax.set_yticklabels( ('Label1', 'Label2', 'Label3', 'Label4', 'Label5') )
ax.yaxis.set_tick_params(length=0,labelbottom=True)

for i, v in enumerate(values1):
    plt.text(v * 0.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

for i, v in enumerate(values2):
    plt.text(v * 1.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

for i, v in enumerate(values3):
    plt.text(v * 2.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)

plt.show()

И код дает мне следующий результат:

Как видите, метки в зеленой и красной частях не выровнены должным образом. Что мне нужно сделать, чтобы исправить эту проблему?

Answer 1

Просто добавьте значение из каждого из предыдущих списков с соответствующим индексом, например:

for i, v in enumerate(values1):
    plt.text(v * 0.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

for i, v in enumerate(values2):
    plt.text(v * 0.45 + values1[i], i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')


for i, v in enumerate(values2):
    plt.text(v * 0.45 + values1[i] + values2[i], i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

Результат:

Answer 2

Коэффициенты 1,45 и 2,45 дадут желаемый результат, только если числа в values1, values2, values3 равны.

Вам необходимо сделать следующее:

Для второго столбца , x = значение первого столбца + 0,45 * значение второго столбца

Для третьего бара , x = значение первого бара + значение второго бара + 0,45 * значение третьего бара

---

Ниже описано, как это можно сделать.

# Use values1[i] + v * 0.45 as the x-coordinate
for i, v in enumerate(values2):
    plt.text(values1[i] + v * 0.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

# Use values1[i] + values2[i] + v * 0.45 as the x-coordinate
for i, v in enumerate(values3):
    plt.text(values1[i] + values2[i] + v * 0.45, i + .145, str(v), color='white', fontweight='bold', fontsize=10, 
             ha='center', va='center')

Answer 3

Я бы заменил все три цикла for i,v на это:

for patch in ax.patches:
    x,y = patch.get_xy()
    w, h = patch.get_width(), patch.get_height()
    ax.text(x+w/2, y+h/2, str(w), 
            ha='center', va='center', 
            color='white', fontweight='bold', fontsize=10, 
           )

Вывод: