cudf.core.column.lists.ListMethods.get#

ListMethods.get(index: int | ColumnLike, default: ScalarLike | ColumnLike | None = None) ParentType[source]#

从列表系列中的每个列表中提取给定索引处的元素。

index 可以是一个整数或一个整数序列。如果 index 是一个整数,则从每个列表中提取位置 index 处的元素。如果 index 是一个序列,则它必须与 Series 的长度相同,并且 index[i] 指定了从 Series 中的第 i 个列表中提取元素的位置。

如果索引超出任何列表的范围,返回 或者,如果提供了,default。因此,此方法永远不会引发 IndexError

Parameters:
indexint or sequence of ints
defaultscalar, optional
Returns:
Series or Index

示例

>>> s = cudf.Series([[1, 2, 3], [3, 4, 5], [4, 5, 6]])
>>> s.list.get(-1)
0    3
1    5
2    6
dtype: int64
>>> s = cudf.Series([[1, 2], [3, 4, 5], [4, 5, 6]])
>>> s.list.get(2)
0    <NA>
1       5
2       6
dtype: int64
>>> s.list.get(2, default=0)
0   0
1   5
2   6
dtype: int64
>>> s.list.get([0, 1, 2])
0   1
1   4
2   6
dtype: int64