Есть ли эквивалент np.where в Polars?Python

Программы на Python
Anonymous
Есть ли эквивалент np.where в Polars?

Сообщение Anonymous »

Polars, есть ли эквивалент np.where? пытаюсь воспроизвести следующий код в полярах.
Если значение выше определенного порогового значения, столбец Is_Acceptable равен 1 или ниже — 0
import polars as pl
import numpy as np

df = pl.DataFrame({"fruit":["orange","apple","mango","kiwi"], "value":[1.0,0.8,0.7,1.2]})

df_pd = df.to_pandas()
df_pd["Is_Acceptable?"] = np.where(df_pd["value"].lt(0.9), 1, 0)
print(pl.from_pandas(df_pd))

shape: (4, 3)
┌────────┬───────┬────────────────┐
│ fruit ┆ value ┆ Is_Acceptable? │
│ --- ┆ --- ┆ --- │
│ str ┆ f64 ┆ i64 │
╞════════╪═══════╪════════════════╡
│ orange ┆ 1.0 ┆ 0 │
│ apple ┆ 0.8 ┆ 1 │
│ mango ┆ 0.7 ┆ 1 │
│ kiwi ┆ 1.2 ┆ 0 │
└────────┴───────┴────────────────┘


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

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