跳过groupby对象的前n项

2uluyalo 于 2021-09-08 发布在 Java

关注(0)|答案(1)|浏览(277)

我已按内容的ID将 Dataframe 分组为 df_grouped = df_input.groupby("ID") 现在，我想跳过/删除groupby对象（df_grouped）的前n个组（“行”），以便“新”groupby对象从n+1开始。下面是一些示例：

| ID | Content |
| ---| --------|
| 3  | just    |
| 4  | a       |
| 6  | short   |
| 10 | sample  |

跳过前2组（n=2）的df_，用类似 df_grouped[2:] ```
| ID | Content |
| ---| --------|
| 6 | short |
| 10 | sample |

跳过前n个组后，数据仍应作为groupby对象存储。

1条答案

先过滤，然后分组

df.loc[df.ID.isin(df.ID.unique().tolist()[2:])].groupby("ID")