Lendenwerte der Füllwerte vor Teilmenge der Werte in Polaren ⇐ Python

[Anonymous]

Ich habe diesen Datenrahmen: < /p>

Code: Select all

import polars as pl

df = pl.DataFrame({'value': [1,2,3,4,5,None, None], 'flag': [0,1,1,1,0,0,0]})
< /code>
┌───────┬──────┐
│ value ┆ flag │
│   --- ┆  --- │
│   i64 ┆  i64 │
╞═══════╪══════╡
│     1 ┆    0 │
│     2 ┆    1 │
│     3 ┆    1 │
│     4 ┆    1 │
│     5 ┆    0 │
│  null ┆    0 │
│  null ┆    0 │
└───────┴──────┘
< /code>
I want to use df.with_columns(pl.col('value').forward_fill())

(oder ähnlich), aber ich möchte nur Werte verwenden, die flag == 1 zum Füllen berechtigt sind. In diesem Beispiel möchte ich, dass Value 4 verwendet werden soll, um die beiden Null -Inträge zu ersetzen (anstelle von 5 ).
┌───────┬──────┐
│ value ┆ flag │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═══════╪══════╡
│ 1 ┆ 0 │
│ 2 ┆ 1 │
│ 3 ┆ 1 │
│ 4 ┆ 1 │
│ 5 ┆ 0 │
│ 4 ┆ 0 │
│ 4 ┆ 0 │
└───────┴──────┘
< /code>
How can one achieve this?