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


Python cudf.DataFrame.take用法及代码示例


用法:

DataFrame.take(indices, axis=0, keep_index=None)

返回一个包含 positions 指定的行的新对象

参数

indicesarray-like

指示要采取哪些位置的整数数组。

keep_index布尔值,默认为真

是否在结果中保留索引。

返回

out系列或 DataFrame 或索引

具有所需行子集的新对象。

例子

系列 >>> s = cudf.Series([‘a’, ‘b’, ‘c’, ‘d’, ‘e’]) >>> s.take([2, 0, 4, 3]) 2 c 0 a 4 e 3 d dtype: object

DataFrame

>>> a = cudf.DataFrame({'a': [1.0, 2.0, 3.0],
...                    'b': cudf.Series(['a', 'b', 'c'])})
>>> a.take([0, 2, 2])
     a  b
0  1.0  a
2  3.0  c
2  3.0  c
>>> a.take([True, False, True])
     a  b
0  1.0  a
2  3.0  c

index

>>> idx = cudf.Index(['a', 'b', 'c', 'd', 'e'])
>>> idx.take([2, 0, 4, 3])
StringIndex(['c' 'a' 'e' 'd'], dtype='object')

相关用法


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