polars.Series.set

Series.set(
    filter: Series,
    value: Any,
) → Series

Задать значения для элементов, отмеченных маской.

Параметры:

filter

Булева маска.

value

Значение, которым заменяются элементы, отмеченные маской.

Примечания

Использование этой функции часто считается антипаттерном, поскольку она может препятствовать оптимизации (проталкиванию предикатов и т. д.). Вместо этого рассмотрите возможность использования pl.when(predicate).then(value).otherwise(self).

Примеры

>>> s = pl.Series("a", [1, 2, 3])
>>> s.set(s == 2, 10)
shape: (3,)
Series: 'a' [i64]
[
    1
    10
    3
]

Лучше реализовать это следующим образом:

>>> s.to_frame().select(
...     pl.when(pl.col("a") == 2).then(10).otherwise(pl.col("a"))
... )
shape: (3, 1)
┌─────────┐
│ literal │
│ ---     │
│ i64     │
╞═════════╡
│ 1       │
│ 10      │
│ 3       │
└─────────┘