Поляры – заполнение нулей в столбце на основе значений из другого ⇐ Python

[Anonymous]

Я пытался изучить Polars, и одна вещь, которую я делаю с Pandas, но до сих пор не мог понять, как это сделать, — это заполнять значениями NULL из столбца, используя словарь, для сравнения значений в другом столбце. Например:

Код: Выделить всё

import polars as pl

df = pl.DataFrame({
"a":[float("nan"),2,3,float("nan"),5],
"b":[1,2,3,4,5],
})

mapping={1:1,
2:4,
3:6,
4:8,
5:10}

df_pd = df.to_pandas()
df_pd["a"] = df_pd["a"].fillna(df_pd["b"].map(mapping))

В результате получится следующий DataFrame:

Код: Выделить всё

shape: (5, 2)
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ f64 ┆ i64 │
╞═════╪═════╡
│ 1.0 ┆ 1   │
│ 2.0 ┆ 2   │
│ 3.0 ┆ 3   │
│ 8.0 ┆ 4   │
│ 5.0 ┆ 5   │
└─────┴─────┘

Итак, мой вопрос: как я могу сделать что-то подобное с помощью Polars? Заранее спасибо!

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