Eu estava procurando uma maneira de anotar minhas barras em um gráfico de barras do Pandas com os valores numéricos arredondados do meu 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
Eu gostaria de obter algo assim:
Tentei com este exemplo de código, mas as anotações estão todas centradas nos x ticks:
>>> 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')
Respostas:
Você obtém isso diretamente dos patches dos eixos:
for p in ax.patches: ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005))
Você vai querer ajustar a formatação da string e os deslocamentos para centralizar as coisas, talvez usar a largura de
p.get_width()
, mas isso deve ajudá-lo a começar. Pode não funcionar com plotagens de barras empilhadas, a menos que você rastreie os deslocamentos em algum lugar.fonte
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')
ax.annotate(str(p.get_width()), (p.get_x() + p.get_width(), p.get_y()), xytext=(5, 10), textcoords='offset points')
ax.annotate(str(int(p.get_width())), (p.get_x() + p.get_width(), p.get_y()), xytext=(-2, 4), textcoords='offset points', horizontalalignment='right')
+1Solução que também lida com os valores negativos com formatação flutuante de amostra.
Ainda precisa de ajustes de compensação.
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))
fonte
O machado nos dá o tamanho da caixa.
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()
para mais clareza ::
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)
esta é a saída de patch.get_bbox () em meu programa.
podemos extrair os detalhes da caixa delimitadora daqui e manipular para nossos requisitos
fonte