你如何删除每组熊猫的某些行[重复]

Posted 2023-02-16

【中文标题】你如何删除每组熊猫的某些行[重复]

【英文标题】：How do you drop certain rows per group pandas [duplicate]

【发布时间】：2021-11-01 13:17:41

【问题描述】：

我有一个数据框，想为每个类别删除某些行。这是数据：

data='GROUP':['A','A','A','B','B','B','B','C','C','C','C','C'],'DATE':['202101','202102','202103','201907','201908','201909',
'201910','202003','202004','202005','202006','202007']
df=pd.DataFrame(data, columns=['GROUP','DATE']) 
         

   GROUP    DATE
0      A  202101
1      A  202102
2      A  202103
3      B  201907
4      B  201908
5      B  201909
6      B  201910
7      C  202003
8      C  202004
9      C  202005
10     C  202006
11     C  202007

我想删除每组第二个日期之后的所有行。换句话说，我想产生这样的效果：

  GROUP    DATE
0     A  202101
1     A  202102
3     B  201907
4     B  201908
7     C  202003
8     C  202004

【参考方案1】：

使用GroupBy.head:

df.groupby('GROUP').head(2)

输出

  GROUP    DATE
0     A  202101
1     A  202102
3     B  201907
4     B  201908
7     C  202003
8     C  202004