Polaren – Füllen von Nullen in einer Spalte basierend auf Werten aus einer anderen ⇐ Python

[Anonymous]

Ich habe versucht, Polarwerte zu lernen, und eine Sache, die ich mit Pandas mache, ist, dass ich immer noch nicht herausgefunden habe, wie man Nullen aus einer Spalte mit einem Wörterbuch füllt, um die Werte in einer anderen Spalte zu vergleichen. Zum Beispiel:

Code: Select all

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))

Dies würde den folgenden DataFrame ergeben:

Code: Select all

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

Meine Frage lautet also: Wie kann ich mit Polars etwas Ähnliches machen? Vielen Dank im Voraus!