Zerlegen und wiederholen Sie Zeilen basierend auf der in einer Polars-Spalte enthaltenen Zahl ⇐ Python

[Anonymous]

In der Eingabe habe ich diesen Datenrahmen:



numberType
NumberInst
Type




1
Keine
Auto


2
1
Bus


3
1
Plane



Ich habe das gemacht, um die Zeilen aufzulösen und zu duplizieren:

Code: Select all

df = df.with_columns(pl.col("NumberInst").alias("numberInstExclude"))
df = df.select(pl.exclude("numberInstExclude").repeat_by("numberType").explode())

Das Ergebnis, das ich haben möchte, ist:



numberType
NumberInst
Type
NumberInstRep




1
Keine
Auto
1


2
1
Bus
1


2
1
Bus
2


3
1
Ebene
1


3
1
Ebene
2


3
1
Ebene
3



Wenn Sie eine Idee haben, wie könnte ich „NumberInstRep“ erhalten?
Was ich tun möchte, ist:

Code: Select all

if NumberInst is None then NumberInstRep = 1
else use numberType to have NumberInstRep

Vielen Dank.