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


Python pyspark Index.take用法及代码示例


本文简要介绍 pyspark.pandas.Index.take 的用法。

用法:

Index.take(indices: Sequence[int]) → IndexOpsLike

返回给定的元素位置的沿轴的索引。

这意味着我们没有根据对象的 index 属性中的实际值进行索引。我们根据元素在对象中的实际位置进行索引。

参数

indices类数组

一个整数数组,指示要采取的位置。

返回

taken与调用者相同的类型

包含从对象中获取的元素的类似数组。

例子

Series

>>> psser = ps.Series([100, 200, 300, 400, 500])
>>> psser
0    100
1    200
2    300
3    400
4    500
dtype: int64
>>> psser.take([0, 2, 4]).sort_index()
0    100
2    300
4    500
dtype: int64

index

>>> psidx = ps.Index([100, 200, 300, 400, 500])
>>> psidx
Int64Index([100, 200, 300, 400, 500], dtype='int64')
>>> psidx.take([0, 2, 4]).sort_values()
Int64Index([100, 300, 500], dtype='int64')

MultiIndex

>>> psmidx = ps.MultiIndex.from_tuples([("x", "a"), ("x", "b"), ("x", "c")])
>>> psmidx  
MultiIndex([('x', 'a'),
            ('x', 'b'),
            ('x', 'c')],
           )
>>> psmidx.take([0, 2])  
MultiIndex([('x', 'a'),
            ('x', 'c')],
           )

相关用法


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