pandas.core.groupby.DataFrameGroupBy.head

DataFrameGroupBy.head(n=5)[source]

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

Аналогично .apply(lambda x: x.head(n)), но возвращает подмножество строк из исходной DataFrame с сохранением исходного индекса и порядка (флаг as_index игнорируется).

Параметры:

n:int

Если положительное: количество записей, которые необходимо включить с начала каждой группы. Если отрицательное: количество записей, которые необходимо исключить с конца каждой группы.

Возвращаемое значение:

Series или DataFrame

Подмножество исходного Series или DataFrame, как определено значением n.

См. также

Series.groupby

Применение функции groupby к Series.

DataFrame.groupby

Применение функции groupby к каждой строке или столбцу DataFrame.

Примеры

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