Python 将数据框中的月度数据分组以显示行小计

发布于10月04日

我有下表，其中Date是索引列:

Date	ColA	ColB
2021-01-01	10	40
2021-01-02	20	15
2022-01-01	10	40
2022-01-02	20	15

我想添加行小计以显示年度合计:

Date	ColA	ColB
2021-01-01	10	40
2021-01-02	20	15
2021 Subtotal	30	55
2022-01-02	20	15
2021-01-02	20	15
2022 Subtotal	40	30

我try 了以下代码:

df.groupby(level='Date').transform("sum")

However I am getting the following error:
DateFormatError: invalid date '2022 Subtotal'

有没有其他方法可以达到类似的效果？

推荐答案

您可以使用:

# grouper for years
# convert to datetime and extract year
# convert to string and add ' Subtotal'
group = (pd.to_datetime(df['Date']).dt.year
         .astype(str)
         .add(' Subtotal')
         )

# perform groupby.sum using the above grouper
# concatenation with original data
# sorting by year to move the subtotals below their respective data
out = (pd.concat(
       [df,
        df.groupby(group)
          .sum().reset_index()])
         .sort_values(by='Date',
                      key=lambda x: x.str.extract('(\d+)', expand=False),
                      kind='stable'
                      )
)

输出:

            Date  ColA  ColB
0     2021-01-01    10    40
1     2021-01-02    20    15
0  2021 Subtotal    30    55
2     2022-01-01    10    40
3     2022-01-02    20    15
1  2022 Subtotal    30    55