python - 使用 DataFrame.plot 在堆积条形图中显示总计和百分比

Question

我的数据框如下所示:

<表类=“s-表”><标题>机场ATA 费用目的地处理自定义总成本 <正文>PRG5992221109520174630491LXU3647151159811595387908AMS4013822356216680441623PRG5992221109520174630491

使用下面的代码可以得到一个堆积条形图:

import pandas as pd

# read the sample markdown table on this page with
df = pd.read_html('https://stackoverflow.com/q/51495982/7758804')[0]

# plot columns without Total Cost
df.iloc[:, :-1].plot(x='Airport', kind='barh', stacked=True, title='Breakdown of Costs', mark_right=True)    

如何在每个堆叠条形图上添加总计(以千 1,000 分隔)？如何添加%对于堆叠条形图中的每个部分？

Answer 1

您可以使用plt.text将信息放置在根据您的数据的位置。

但是，如果您的条形非常小，则可能需要进行一些调整才能看起来完美。

df_total = df['Total Cost']
df = df.iloc[:, 0:4]
df.plot(x = 'Airport', kind='barh',stacked = True, title = 'Breakdown of Costs', mark_right = True)

df_rel = df[df.columns[1:]].div(df_total, 0)*100

for n in df_rel:
    for i, (cs, ab, pc, tot) in enumerate(zip(df.iloc[:, 1:].cumsum(1)[n], df[n], df_rel[n], df_total)):
        plt.text(tot, i, str(tot), va='center')
        plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va='center', ha='center')

编辑:一些为了更好的可读性而任意的想法:

将总值向右移动，使用 45° 旋转文本:

    plt.text(tot+10000, i, str(tot), va='center')
    plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va='center', ha='center', rotation=45)

在顶部和底部对齐文本之间切换:

va = ['top', 'bottom']
va_idx = 0
for n in df_rel:
    va_idx = 1 - va_idx
    for i, (cs, ab, pc, tot) in enumerate(zip(df.iloc[:, 1:].cumsum(1)[n], df[n], df_rel[n], df_total)):
        plt.text(tot+10000, i, str(tot), va='center')
        plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va=va[va_idx], ha='center')

仅标记 10% 或更多的条形:

if pc >= 10:
    plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va='center', ha='center')

...或者仍然打印它们，但是垂直:

if pc >= 10:
    plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va='center', ha='center')
else:
    plt.text(cs - ab/2, i, str(np.round(pc, 1)) + '%', va='center', ha='center', rotation=90)