Панды: как сделать предварительный просмотр нескольких строк, отсортированных по определенным столбцам

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

Спасибо

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