pandas.core.groupby.GroupBy.head

GroupBy.head(n=5)

返回每个组的前n行。

基本上等同于.apply（lambda x： x.head（n）），除了忽略as_index标志。

也可以看看

pandas.Series.groupby，pandas.DataFrame.groupby，pandas.Panel.groupby

例子

>>> df = DataFrame([[1, 2], [1, 4], [5, 6]],
                   columns=['A', 'B'])
>>> df.groupby('A', as_index=False).head(1)
   A  B
0  1  2
2  5  6
>>> df.groupby('A').head(1)
   A  B
0  1  2
2  5  6