pyspark.pandas.groupby.SeriesGroupBy.nlargest

SeriesGroupBy.nlargest(n=5)

Return the first n rows ordered by columns in descending order in group.

Return the first n rows with the smallest values in columns, in descending order. The columns that are not specified are returned as well, but not used for ordering.

Parameters

nint

Number of items to retrieve.

See also

pyspark.pandas.Series.nlargest

pyspark.pandas.DataFrame.nlargest

Examples

>>> df = ps.DataFrame({'a': [1, 1, 1, 2, 2, 2, 3, 3, 3],
...                    'b': [1, 2, 2, 2, 3, 3, 3, 4, 4]}, columns=['a', 'b'])

>>> df.groupby(['a'])['b'].nlargest(1).sort_index()  
a
1  1    2
2  4    3
3  7    4
Name: b, dtype: int64