选择满足条件的行

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