web-dev-qa-db-ja.com

Pandasバープロット上の値でバーに注釈を付けます

私は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

私はこのようなものを手に入れたいです：

bar plot annotation example

このコードサンプルを試しましたが、注釈はすべて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')
pythonmatplotlibplotpandasdataframe
56
leroygr

軸のパッチから直接取得します。

_In [35]: for p in ax.patches:
    ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005))
_

文字列の書式設定とオフセットを微調整して、物事を中央に配置します。おそらくp.get_width()の幅を使用しますが、それで開始できます。オフセットをどこかで追跡しない限り、積み上げ棒グラフでは機能しない場合があります。

90
TomAugspurger

サンプルのフロート形式で負の値も処理するソリューション。

依然としてオフセットの調整が必要です。

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

value labeled bar plot

23
tworec