Фильтрация строк данных на основе количества

Question

У меня есть таблица типа

Fruit   Month
----------------
A   Jan
A   Feb
A   Mar 
B   Sep

Я хочу отфильтровать таблицу таким образом, чтобы я получал только верхние n строк, скажем, две для каждого фрукта. Пример

Fruit   Month
----------------
A   Jan
A   Feb 
B   Sep

Answer 1

Вы можете просто использовать groupby() и head():

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

Выходы:

  Fruit Month
0     A   Jan
1     A   Feb
3     B   Sep