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


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