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


Python cudf.core.column.lists.ListMethods.take用法及代码示例


用法:

ListMethods.take(lists_indices: Any) → Union[cudf.core.series.Series, cudf.core.index.GenericIndex]

根据给定的索引收集列表元素。

参数

lists_indices: Series-like of lists

指定从每一行收集的内容

返回

系列或索引

例子

>>> s = cudf.Series([[1, 2, 3], None, [4, 5]])
>>> s
0    [1, 2, 3]
1         None
2       [4, 5]
dtype: list
>>> s.list.take([[0, 1], [], []])
0    [1, 2]
1      None
2        []
dtype: list

相关用法


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