Python pyspark DataFrame.idxmin用法及代码示例

本文简要介绍 pyspark.pandas.DataFrame.idxmin 的用法。

用法: DataFrame.idxmin(axis: Union[int, str] = 0) → Series

请求轴上第一次出现最小值的返回索引。 NA/空值被排除在外。

注意

此 API 使用 to_pandas() 收集所有具有最小值的行，因为我们假设具有最小值的行数通常很少。

参数：

axis：0 或 ‘index’: 目前只能设置为0。

Series

例子：

>>> psdf = ps.DataFrame({'a': [1, 2, 3, 2],
...                     'b': [4.0, 2.0, 3.0, 1.0],
...                     'c': [300, 200, 400, 200]})
>>> psdf
   a    b    c
0  1  4.0  300
1  2  2.0  200
2  3  3.0  400
3  2  1.0  200

>>> psdf.idxmin()
a    0
b    3
c    1
dtype: int64

对于Multi-column 索引

>>> psdf = ps.DataFrame({'a': [1, 2, 3, 2],
...                     'b': [4.0, 2.0, 3.0, 1.0],
...                     'c': [300, 200, 400, 200]})
>>> psdf.columns = pd.MultiIndex.from_tuples([('a', 'x'), ('b', 'y'), ('c', 'z')])
>>> psdf
   a    b    c
   x    y    z
0  1  4.0  300
1  2  2.0  200
2  3  3.0  400
3  2  1.0  200

>>> psdf.idxmin()
a  x    0
b  y    3
c  z    1
dtype: int64

相关用法

注：本文由纯净天空筛选整理自spark.apache.org大神的英文原创作品 pyspark.pandas.DataFrame.idxmin。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

例子：