选择满足条件的行

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查询，但是无法编写正确的查询。你能帮忙吗 ？也许“ for循环”将是更好的解决方案……

谢谢所有！

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