polars.arg_where

polars.arg_where(
    condition: Expr | Series,
    *,
    eager: bool = False,
) → Expr | Series

Возвращает индексы, для которых condition имеет значение True.

Параметры:

condition

Логическое выражение для вычисления

eager

Вычислить немедленно и вернуть Series; для этого заданное условие должно быть Series. Если установлено значение False (по умолчанию), вместо этого возвращается выражение.

См. также

Series.arg_true

Возвращает индексы, в которых значение Series равно True

Примеры

>>> df = pl.DataFrame({"a": [1, 2, 3, 4, 5]})
>>> df.select(
...     [
...         pl.arg_where(pl.col("a") % 2 == 0),
...     ]
... ).to_series()
shape: (2,)
Series: 'a' [u32]
[
    1
    3
]