Выберите строки, которые удовлетворяют условиям

Question

Мне нужно написать запрос, который получит название компании и покажет только 2 строки, после чего проверит следующую компанию и покажет еще 2 строки.Допустим, что df выглядит так, как показано ниже:

x     y     y      name
1      2    3       ammazon
4     5     6       ammazon
7     8     9       ammazon
9     8     7       google
6     5     4       google
3     2     1       google 

Таким образом, результат должен быть таким:

 x     y     y      name
1      2    3       ammazon
4     5     6       ammazon
9     8     7       google
6     5     4       google

Я пытался использовать SQL-запрос, но не смог написать правильный.Не могли бы вы помочь?Или, возможно, «для цикла» было бы лучшим решением ... что-нибудь

Спасибо всем!

Answer 1

groupby с head

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

   x  y  y     name
0  1  2  3  ammazon
1  4  5  6  ammazon
3  9  8  7   google
4  6  5  4   google