Verwenden Sie ein Ausdruckswörterbuch, um zeilenweise basierend auf einer Spalte in Polarzahlen zu berechnen ⇐ Python

[Guest]

Ich möchte ein Ausdruckswörterbuch verwenden, um Berechnungen für eine neue Spalte durchzuführen.
Ich habe diesen Polars-Datenrahmen:

Code: Select all

df=pl.DataFrame(
"col1": ["a", "b", "a"],
"x": [1,2,3],
"y": [2,2,5]
)

Und ich habe ein Ausdruckswörterbuch:

Code: Select all

expr_dict = {
"a": pl.col("x") * pl.col("y"),
"b": pl.col("x"),
}

Ich möchte eine Spalte erstellen, in der jeder Wert basierend auf einer Eingabe in einer anderen Spalte berechnet wird, weiß aber nicht wie. Ich möchte ein Ergebnis wie dieses haben:

Code: Select all

>>> df.with_columns(r=pl.col("col1").apply(lambda x: expr_dict[X])
>>> shape: (3, 3)
┌──────┬─────┬─────┬─────┐
│ col1 ┆ x   ┆ y   ┆ r   │
│ ---  ┆ --- ┆ --- ┆ --- │
│ str  ┆ i64 ┆ i64 ┆ i64 │
╞══════╪═════╪═════╪═════╡
│ a    ┆ 1   ┆ 2   ┆ 2   │
│ b    ┆ 2   ┆ 2   ┆ 4   │
│ a    ┆ 3   ┆ 5   ┆ 15  │
└──────┴─────┴─────┴─────┘

Ist das möglich?