Pandas 数据框选择性数据清理 post groupby

Question

我是 Pandas 的新手，我想知道如何通过仅提取行的一部分来清理数据。假设我有一个数据框，如下所示:

column1      date    key
A            2016    SB
A            2017    B
B            2015    SB
C            2014    SB
C            2014    PB
C            2015    B
C            2016    SB

我如何清理数据，以便对于每个相同的 column1 值，我只提取前两行值并忽略其余的(例如在 C 值上，只有 2014 SB 和 2014 PB 是我得到的)？

column1      date    key
A            2016    SB
A            2017    B
B            2015    SB
C            2014    SB
C            2014    PB

谢谢

Answer 1

您需要 GroupBy.head , 也检查 docs :

df = df.groupby('column1').head(2)
print (df)
  column1  date key
0       A  2016  SB
1       A  2017   B
2       B  2015  SB
3       C  2014  SB
4       C  2014  PB