pandas - Pandas：如何预览在某些列下排序的几行

Question

如果我有一个数据框（df）：

Year Rate
2001 10
2001 3 
2001 5
2001 3
2001 6
2002 2
2002 7
2002 4
2002 9
2002 8
...  ...
2018 8
2018 6
2018 4
2018 6
2018 5

我如何获得一个仅显示每年前 2 行的数据框，例如：

Year Rate
2001 10
2001 3 
2002 2
2002 7
...  ...
2018 8
2018 6

谢谢

标签： pandas

Answer 1

使用GroupBy.head：

df1 = df.groupby('Year').head(2)
print (df1)
    Year  Rate
0   2001    10
1   2001     3
5   2002     2
6   2002     7
10  2018     8
11  2018     6