Verwenden Sie ein Ausdruckswörterbuch, um zeilenweise basierend auf einer Spalte in Polarzahlen zu berechnen
Posted: 28 Dec 2024, 19:10
Ich möchte ein Ausdruckswörterbuch verwenden, um Berechnungen für eine neue Spalte durchzuführen.
Ich habe diesen Polars-Datenrahmen:
Und ich habe ein Ausdruckswörterbuch:
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:
Ist das möglich?
Ich habe diesen Polars-Datenrahmen:
Code: Select all
df=pl.DataFrame(
"col1": ["a", "b", "a"],
"x": [1,2,3],
"y": [2,2,5]
)
Code: Select all
expr_dict = {
"a": pl.col("x") * pl.col("y"),
"b": pl.col("x"),
}
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 │
└──────┴─────┴─────┴─────┘