如何对Pandas已分组DataFrame按组内新增聚合列完成组内排序

实现方案 #

你可以通过以下两种常用方式实现需求，均能得到预期的分组排序结果：

方法1：多规则联合排序（性能更优，代码简洁） #

已经完成聚合得到grp_df后，直接指定排序规则即可：

# pandas 1.1.0及以上版本可直接按索引+列排序
res = grp_df.sort_values(by=['A', 'ctr'], ascending=[True, False], level=0)

如果是更低版本的pandas，可通过临时重置索引实现：

res = grp_df.reset_index()\
            .sort_values(by=['A', 'ctr'], ascending=[True, False])\
            .set_index(['A', 'B'])

方法2：分组后组内单独排序 #

如果需要对每个分组做额外自定义处理，可基于聚合结果按A分组后再单独排序：

# group_keys=False避免生成多余的分组索引层级
res = grp_df.groupby(level='A', group_keys=False)\
            .apply(lambda df: df.sort_values('ctr', ascending=False))

两种方法执行后输出res即可得到你给出的预期结果。

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