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

Question

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

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

Answer 1

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

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