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

Question

我有下表，其中 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

我尝试了以下代码：

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

但是我收到以下错误：
DateFormatError：无效日期“2022 小计”

有什么替代方法可以达到类似的结果？

Answer 1

您可以使用：

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

输出：

            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