當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python pyspark GroupBy.idxmax用法及代碼示例

本文簡要介紹 pyspark.pandas.groupby.GroupBy.idxmax 的用法。

用法:

GroupBy.idxmax(skipna: bool = True) → FrameLike

返回組中請求軸上第一次出現最大值的索引。 NA/空值被排除在外。

參數：

skipna：布爾值，默認 True

排除 NA/空值。如果整行/列為 NA，則結果將為 NA。

例子：

>>> df = ps.DataFrame({'a': [1, 1, 2, 2, 3],
...                    'b': [1, 2, 3, 4, 5],
...                    'c': [5, 4, 3, 2, 1]}, columns=['a', 'b', 'c'])
>>> df.groupby(['a'])['b'].idxmax().sort_index() 
a
1  1
2  3
3  4
Name: b, dtype: int64
>>> df.groupby(['a']).idxmax().sort_index() 
   b  c
a
1  1  0
2  3  2
3  4  4

相關用法


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