pandas.core.groupby.GroupBy.head

GroupBy.head(self, n=5) [source]

Возвращает первые n строк каждой группы.

По существу эквивалентно .apply(lambda x: x.head(n)), за исключением игнорирования флага as_index.

Возвращает:	Серия или DataFrame

См. также

Series.groupby

DataFrame.groupby

Примеры

>>> df = pd.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