将每个组的小计添加到数据框中

Question

我在简洁地编写代码时遇到了一些麻烦（我已经有了一个可用的代码，但理想情况下我想尽可能地缩短它，因为它又大又乱）

我有一个看起来像这样的数据框

Group	Value
A	2
A	4
Total A
B	5
B	3
Total B

是否有一种简洁的方法来汇总并填写每个组的 Total... 行，以便我得到这样的数据框？

Group	Value
A	2
A	4
Total A	6
B	5
B	3
Total B	8

谢谢！

Answer 1

您可以将 groupby 函数与 mean 函数一起使用：

import pandas as pd
df = pd.DataFrame(data=[('A', 2), ('A',4), ('B', 5), ('B', 3)], columns=['Group', 'Value'])
df_sum = df.groupby(by='Group').mean()
print(df_sum)