当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


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

用法:

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

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

注意

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

参数

axis0 或 ‘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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。