polars.Expr.gather

Expr.gather(
    indices: int | Sequence[int] | IntoExpr | Series | np.ndarray[Any,
    Any],
    *,
    null_on_oob: bool = False,
) → Expr

Получить значения по индексу.

движок:В памятиПотоковая обработка

Параметры:

indices

Выражение, результатом которого является Series с типом UInt32.

null_on_oob

Поведение, если индекс выходит за границы:

• True -> установить для результата значение null
• False -> вызвать ошибку

Возвращает:

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]   │
└───────┴───────────┘

Используйте null_on_oob=True, чтобы возвращать null для индексов, выходящих за границы.

>>> df = pl.DataFrame({"a": [1, 2, 3]})
>>> df.select(pl.col("a").gather([0, 1, 10], null_on_oob=True))
shape: (3, 1)
┌──────┐
│ a    │
│ ---  │
│ i64  │
╞══════╡
│ 1    │
│ 2    │
│ null │
└──────┘