设置as_index=False后分组DataFrame索引未重置,该如何修复?
解决分组后DataFrame索引未重置的问题
问题原因
groupby的as_index=False参数作用是不让分组键(如这里的BL)成为结果DataFrame的索引,但不会重置每个分组内部保留的原DataFrame行索引。你看到的4、5、10是原数据里的行号,所以这个参数无法满足重置分组内索引的需求。
解决方法
方法一:循环内对每个分组调用reset_index
遍历分组时,对每个子DataFrame单独重置索引,用drop=True删除原索引列:
for name, df_group in df.groupby('BL'): df_group = df_group.reset_index(drop=True) print(df_group)
执行后每个分组的索引会从0开始连续编号,输出示例:
Num BL Home Requester 0 16986 140080043863936 5 5 1 16987 140080043863936 0 5 2 16986 140080043863936 7 5
方法二:用apply批量处理所有分组
如果不需要逐组遍历操作,可以直接通过apply批量重置所有分组的索引,搭配group_keys=False避免生成多级索引:
df_reset = df.groupby('BL', group_keys=False).apply(lambda x: x.reset_index(drop=True)) print(df_reset)
内容的提问来源于stack exchange,提问作者mahmood
相关产品推荐
相关产品推荐

