python - 如何仅注释堆积条形图的一个类别

Question

我有一个数据框示例，如下所示。

data = {'Date':['2021-07-18','2021-07-19','2021-07-20','2021-07-21','2021-07-22','2021-07-23'],
    'Invalid':["NaN", 1, 1, "NaN", "NaN", "NaN"],
    'Negative':[23, 24, 17, 24, 20, 23],
    'Positive':["NaN", 1, 1, 1, "NaN", 1]}

df_sample = pd.DataFrame(data) 
df_sample

下面给出了显示堆叠条形图的代码以及由它生成的图形。

temp = Graph1_df.set_index(['Dates', 'Results']).sort_index(0).unstack()
temp.columns = temp.columns.get_level_values(1)

f, ax = plt.subplots(figsize=(20, 5))
temp.plot.bar(ax=ax, stacked=True, width = 0.3, color=['blue','green','red'])
ax.title.set_text('Total Test Count vs Dates') 


plt.show()

使用上面的代码或任何新方法，我只想在图表上显示“正”的值。
注意:数据框片段中的第三列是“正”列。
任何帮助是极大的赞赏。
谢谢。

Answer 1

使用 pandas.DataFrame.plot 绘图与 kind='bar'

使用 .bar_label 添加注释

看到这个 answer其他与 .bar_label 相关的链接和选项

堆积条形图分别根据列和行的顺序从左到右和从下到上绘制。

自 'Positive'是列索引 2，我们只想要 i == 2 的标签

在 pandas 1.3.0 中测试并且需要 matplotlib >=3.4.2和 python >=3.8

labels的列表推导式使用赋值表达式，:= ，仅可从 python 3.8 获得

labels = [f'{v.get_height():.0f}' if ((v.get_height()) > 0) and (i == 2) else '' for v in c]是不带 := 的选项

.bar_label仅可从 matplotlib 3.4.2 获得

此 answer显示如何为 matplotlib <3.4.2 添加注释

import pandas as pd
import numpy as np  # used for nan

# test dataframe
data = {'Date':['2021-07-18','2021-07-19','2021-07-20','2021-07-21','2021-07-22','2021-07-23'],
    'Invalid':[np.nan, 1, 1, np.nan, np.nan, np.nan],
    'Negative':[23, 24, 17, 24, 20, 23],
    'Positive':[np.nan, 1, 1, 1, np.nan, 1]}

df = pd.DataFrame(data)

# convert the Date column to a datetime format and use the dt accessor to get only the date component
df.Date = pd.to_datetime(df.Date).dt.date

# set Date as index
df.set_index('Date', inplace=True)

# create multi-index column to match OP image
top = ['Size']
current = df.columns
df.columns = pd.MultiIndex.from_product([top, current], names=['', 'Result'])

# display(df)
              Size                  
Result     Invalid Negative Positive
Date                                
2021-07-18     NaN       23      NaN
2021-07-19     1.0       24      1.0
2021-07-20     1.0       17      1.0
2021-07-21     NaN       24      1.0
2021-07-22     NaN       20      NaN
2021-07-23     NaN       23      1.0

# reset the top index to a column
df = df.stack(level=0).rename_axis(['Date', 'Size']).reset_index(level=1)

# if there are many top levels that are reset as a column, then select the data to be plotted
sel = df[df.Size.eq('Size')]

# plot
ax = sel.iloc[:, 1:].plot(kind='bar', stacked=True, figsize=(20, 5), title='Total Test Count vs Dates', color=['blue','green','red'])

# add annotations
for i, c in enumerate(ax.containers):
    
    # format the labels
    labels = [f'{w:.0f}' if ((w := v.get_height()) > 0) and (i == 2) else '' for v in c]
    
    # annotate with custom labels
    ax.bar_label(c, labels=labels, label_type='center', fontsize=10)

    # pad the spacing between the number and the edge of the figure
    ax.margins(y=0.1)