cudf.core.column.lists.ListMethods.take

ListMethods.take(lists_indices: Any) → Series | Index

Collect list elements based on given indices.

Parameters:

lists_indices: Series-like of lists

Specifies what to collect from each row

Returns:

Series or Index

Examples

>>> 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