Groupby Pandas 数据框并根据排名有条件地丢弃值

Question

我有一个这样的数据框：

Col1 | Col2 | Col3
 a   |   8  |  9
 a   |   3  |  7
 a   |   1  |  3
 a   |   0  |  8
 b   |   6  |  18
 b   |   2  |  6

如果可能的话，我想通过对 Col1 进行分组来删除除 Col2 的第 2 和第 3 个最高值之外的所有内容

所需的输出：

Col1 | Col2 | Col3
 a   |   3  |  7
 a   |   1  |  3
 b   |   2  |  6

Answer 1

可以使用cumcount

df[df.groupby('Col1').cumcount().isin([1,2])]
Out[423]: 
  Col1  Col2  Col3
1    a     2     7
2    a     1     3
5    b     2     6

更多信息：

df.groupby('Col1').cumcount()
Out[435]: 
0    0
1    1
2    2
3    3
4    0
5    1
dtype: int64