Pandas 막대 그림의 값으로 막대에 주석 달기


89

내 DataFrame의 둥근 숫자 값으로 Pandas 막대 그림에서 막대에 주석을 달 수있는 방법을 찾고있었습니다.

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

Tom이 성공했지만 여기에 더 복잡한 솔루션이 있습니다. stackoverflow.com/questions/19917587/…
Paul H

답변:


148

축의 패치에서 직접 가져옵니다.

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

문자열 형식과 오프셋을 조정하여 중심에 맞추고 싶을 것입니다. 아마도에서 너비를 사용할 수도 p.get_width()있지만 시작해야합니다. 오프셋을 어딘가에서 추적하지 않으면 누적 막대 플롯에서 작동하지 않을 수 있습니다.


28
감사합니다 @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 2014-08-22

1
수평 막대에 대한 유사한 솔루션이 있습니까 : kind = barh?
Nickpick 2015 년

4
나는 이것이 barh를 위해 일하는 것을 발견했다 :ax.annotate(str(p.get_width()), (p.get_x() + p.get_width(), p.get_y()), xytext=(5, 10), textcoords='offset points')
Kamil Sindi

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

값 레이블이있는 막대 그림


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 () 출력입니다.
여기에서 경계 상자 세부 정보를 추출하고 요구 사항에 맞게 조작 할 수 있습니다.

당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.