如何在pandas df列上使用聚合并保留原始df

Question

我想将聚合应用于宠物年龄列，但保留应用此聚合时消失的原始df列。从图像中，如何在应用聚合时将其他列添加到生成的df中。

我还可以将其他列转换为数字类型，或者将其与原始列合并？谢谢！

Answer 1

只需使用transform进行内联聚合或任何返回类似索引对象的特定于组的计算：

merge['pet_age_group_mean'] = merge.groupby('user_id')['pet_age'].transform('mean')