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


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。