python - 使用返回超过 1 行的聚合函数对熊猫数据框进行分组？

Question

假设我有一个如下所示的数据框：

   grp  x  y    z
a    1  6  4  3.0
b    2  5  5  4.0
c    1  4  6  5.0
d    2  3  7  6.0
e    1  2  8  7.0
f    2  1  9  NaN

我想按grp列分组并从每组中选择前两行。所以这就是我的期望：

   grp  x  y    z
a    1  6  4  3.0
c    1  4  6  5.0
b    2  5  5  4.0
d    2  3  7  6.0

我如何用熊猫做到这一点？

标签： pythonpandasdataframepandas-groupby