Python pyspark SeriesGroupBy.nlargest用法及代碼示例

本文簡要介紹 pyspark.pandas.groupby.SeriesGroupBy.nlargest 的用法。

用法: SeriesGroupBy.nlargest(n: int = 5) → pyspark.pandas.series.Series

返回組中按列降序排列的前 n 行。

按降序返回列中具有最小值的前 n 行。未指定的列也會返回，但不用於排序。

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

相關用法

注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.pandas.groupby.SeriesGroupBy.nlargest。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。