polars.Expr.gather

Expr.gather(

indices: int | Sequence[int] | IntoExpr | Series | np.ndarray[Any, Any],

) → Expr

按索引取值。

Parameters:

indices

一个导致UInt32类型Series的表达式。

Returns:

Expr

相同数据类型的表达式。

另请参阅

Expr.get

获取单个值

示例

>>> df = pl.DataFrame(
...     {
...         "group": [
...             "one",
...             "one",
...             "one",
...             "two",
...             "two",
...             "two",
...         ],
...         "value": [1, 98, 2, 3, 99, 4],
...     }
... )
>>> df.group_by("group", maintain_order=True).agg(
...     pl.col("value").gather([2, 1])
... )
shape: (2, 2)
┌───────┬───────────┐
│ group ┆ value     │
│ ---   ┆ ---       │
│ str   ┆ list[i64] │
╞═══════╪═══════════╡
│ one   ┆ [2, 98]   │
│ two   ┆ [4, 99]   │
└───────┴───────────┘