如何在DataFrame中按model列添加分组求和列

现有一个包含model和qty列的DataFrame，数据如下：

import pandas as pd
df = pd.DataFrame({'model':['a','a','a','b', 'b','c','c','d','d','d'],
   'qty':[1,2,1,3,1,5, 2,3,4,5]})

需要在DataFrame末尾新增一列，该列显示对应model分组下qty的总和——即每个相同model的行，都显示该组的qty合计值。

可以使用pandas的groupby结合transform方法实现，该方法会将分组计算的结果映射回原DataFrame的每一行：

# 添加求和列
df['sum_qty'] = df.groupby('model')['qty'].transform('sum')

# 查看最终结果
print(df)

执行后得到的结果如下：

内容的提问来源于stack exchange，提问作者ghost_like