Измените значения нескольких столбцов на ноль или 0 в зависимости от другого условия столбца в Polars [дубликат]Python

Программы на Python
Anonymous
Измените значения нескольких столбцов на ноль или 0 в зависимости от другого условия столбца в Polars [дубликат]

Сообщение Anonymous »

Как присвоить одно и то же значение нескольким столбцам на основе другого условия столбца в полярах Python
import polars as pl

df = pl.DataFrame({
"A": ["a", "b", "a", "c"],
"B": [23,45,60,50],
"C": [11, 26, 63, 84]
})

┌─────┬─────┬─────┐
│ A ┆ B ┆ C │
│ --- ┆ --- ┆ --- │
│ str ┆ i64 ┆ i64 │
╞═════╪═════╪═════╡
│ a ┆ 23 ┆ 11 │
│ b ┆ 45 ┆ 26 │
│ a ┆ 60 ┆ 63 │
│ c ┆ 50 ┆ 84 │
└─────┴─────┴─────┘

На основе столбца A, если значение равно «a», то значение для столбцов B и C должно быть установлено как 0.
Ожидаемый результат:
┌─────┬─────┬─────┐
│ A ┆ B ┆ C │
│ --- ┆ --- ┆ --- │
│ str ┆ i64 ┆ i64 │
╞═════╪═════╪═════╡
│ a ┆ 0 ┆ 0 │
│ b ┆ 45 ┆ 26 │
│ a ┆ 0 ┆ 0 │
│ c ┆ 50 ┆ 84 │
└─────┴─────┴─────┘


Подробнее здесь: https://stackoverflow.com/questions/759 ... ndition-in

Вернуться в «Python»