Аннотирование столбцов со значениями на диаграммах столбцов Pandas

89

Я искал способ аннотировать свои столбцы на графике столбцов Pandas округленными числовыми значениями из моего DataFrame.

>>> df=pd.DataFrame({'A':np.random.rand(2),'B':np.random.rand(2)},index=['value1','value2'] )         
>>> df
                 A         B
  value1  0.440922  0.911800
  value2  0.588242  0.797366

Хотелось бы получить примерно такое:

пример аннотации гистограммы

Я пробовал использовать этот образец кода, но все аннотации сосредоточены на отметках x:

>>> ax = df.plot(kind='bar') 
>>> for idx, label in enumerate(list(df.index)): 
        for acc in df.columns:
            value = np.round(df.ix[idx][acc],decimals=2)
            ax.annotate(value,
                        (idx, value),
                         xytext=(0, 15), 
                         textcoords='offset points')
python matplotlib plot pandas dataframe Leroygr
источник
Том справился, но у меня есть более сложное решение: stackoverflow.com/questions/19917587/…
Paul H

Ответы:

149

Вы получаете его прямо из нашивок топоров:

for p in ax.patches:
    ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005))

Вы захотите настроить форматирование строки и смещения, чтобы все было по центру, возможно, используйте ширину от p.get_width(), но это должно помочь вам начать. Это может не работать с составными столбчатыми диаграммами, если вы где-то не отслеживаете смещения.

TomAugspurger
источник
29
Спасибо @TomAugsPurger, сработало следующее: for p in ax.patches: ax.annotate(np.round(p.get_height(),decimals=2), (p.get_x()+p.get_width()/2., p.get_height()), ha='center', va='center', xytext=(0, 10), textcoords='offset points')
leroygr
2
Другой вопрос: как бы вы поступили с барами с отрицательными значениями? В приведенном выше коде метки всегда имеют положительные координаты, даже если значение столбца отрицательное.
leroygr
1
Есть ли подобное решение для турников: kind = barh?
Nickpick 05
4
Я обнаружил, что это работает для barh:ax.annotate(str(p.get_width()), (p.get_x() + p.get_width(), p.get_y()), xytext=(5, 10), textcoords='offset points')
Камил Синди
2
Спасибо @capitalistpug. Я обнаружил, что добавление горизонтального выравнивания сделало ваше еще лучше. ax.annotate(str(int(p.get_width())), (p.get_x() + p.get_width(), p.get_y()), xytext=(-2, 4), textcoords='offset points', horizontalalignment='right') +1
DaveL17
31

Решение, которое также обрабатывает отрицательные значения с форматированием образца с плавающей запятой.

По-прежнему нужно настроить смещения.

df=pd.DataFrame({'A':np.random.rand(2)-1,'B':np.random.rand(2)},index=['val1','val2'] )
ax = df.plot(kind='bar', color=['r','b']) 
x_offset = -0.03
y_offset = 0.02
for p in ax.patches:
    b = p.get_bbox()
    val = "{:+.2f}".format(b.y1 + b.y0)        
    ax.annotate(val, ((b.x0 + b.x1)/2 + x_offset, b.y1 + y_offset))

гистограмма с обозначением значений

tworec
источник
0

Топор дает нам размер коробки. 

x_position=##define a value
y_position=##define a value
for patch in ax.patches:
    b= patch.get_bbox()
    y_value=b.y1-b.y0
    ax.annotate(y_value, "x_position" , "y_position"))
plt.show()

для большей ясности ::
Bbox (x0 = 3,75, y0 = 0,0, x1 = 4,25, y1 = 868,0)
Bbox (x0 = 4,75, y0 = 0,0, x1 = 5,25, y1 = 868,0)
Bbox (x0 = 5,75, y0 = 0,0 , x1 = 6,25, y1 = 1092,0)
Bbox (x0 = 6,75, y0 = 0,0, x1 = 7,25, y1 = 756,0)
Bbox (x0 = 7,75, y0 = 0,0, x1 = 8,25, y1 = 756,0)
Bbox (x0 = 8,75 , y0 = 0,0, x1 = 9,25, y1 = 588,0)
Bbox (x0 = 3,75, y0 = 868,0, x1 = 4,25, y1 = 3724,0)
Bbox (x0 = 4,75, y0 = 868,0, x1 = 5,25, y1 = 3528,0)
Bbox (x0 = 5,75, y0 = 1092,0, x1 = 6,25, y1 = 3948,0)
Bbox (x0 = 6,75, y0 = 756,0, x1 = 7,25, y1 = 2884,0)
Bbox (x0 = 7,75, y0 = 756,0, x1 = 8,25, y1 = 3024,0)
Bbox (x0 = 0,75, y0 = 4004,0, x1 = 1,25, y1 = 4396,0)
Bbox (x0 = 1,75, y0 = 3668,0, x1 = 2,25, y1 = 4060,0)
Bbox (x0 = 2,75, y0 = 3864,0, x1 = 3,25, y1 = 4060,0)

это результат работы patch.get_bbox () в моей программе.
мы можем извлечь детали ограничивающей рамки отсюда и манипулировать в соответствии с нашим требованием

замораживание кода
источник