xorbits.pandas.groupby.SeriesGroupBy.head#

SeriesGroupBy.head(n=5)#

Return first n rows of each group.

Similar to .apply(lambda x: x.head(n)), but it returns a subset of rows from the original DataFrame with original index and order preserved (as_index flag is ignored).

Parameters: n (int) – If positive: number of entries to include from start of each group. If negative: number of entries to exclude from end of each group.
Returns: Subset of original Series or DataFrame as determined by n.
Return type: Series or DataFrame

See also

Series.groupby: Apply a function groupby to a Series.
DataFrame.groupby: Apply a function groupby to each row or column of a DataFrame.

Examples

>>> 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

This docstring was copied from pandas.core.groupby.generic.SeriesGroupBy.