首页 > 解决方案 > 根据列对数据框进行分组

问题描述

df:
      AAAA   BBBB  CCCC
1111  17878    2    2
2222  17878    2    3
3333  11777    2    4
4444  11888    5    5

我在下面有一个系列:

ser:

17878
11777

我希望我的 df 看起来像这样:

df:
      AAAA   BBBB  CCCC
1111  17878    2    2
2222  17878    2    3
3333  11777    2    4

这是我找到的代码,但不适用于系列

df.groupby(['AAAA']).get_group(ser)

标签: pythonpandas

解决方案


这样的事情可能会奏效-

df = df[df['AAAA'].isin(ser.tolist())]

推荐阅读