Как найти последнее ненулевое значение перед текущей строкой в ​​Polars? ⇐ Python

[Anonymous]

Я хотел бы выполнить следующее:
Ввод:

Код: Выделить всё

import polars as pl

df = pl.DataFrame({
"a": [1,15,None,20,None]
})

Вывод:

Код: Выделить всё

expected = pl.from_repr("""
┌──────┬──────┐
│ a    ┆ b    │
│ ---  ┆ ---  │
│ i64  ┆ i64  │
╞══════╪══════╡
│ 1    ┆ 0    │
│ 15   ┆ 14   │ # b = 15 - 1
│ null ┆ null │
│ 20   ┆ 5    │ # b = 20 - 15
│ null ┆ null │
└──────┴──────┘
""")

Итак, что он делает:

• Если значение «A» равно нулю, то значение B (выходной столбец) также равно нулю.
• Если «A» имеет какое-то значение, извлеките последнее ненулевое значение в «A», а затем вычтите текущее значение в «A» из предыдущего ненулевого значения

Я попробовал следующий вопрос:

• Как выбрать последнее ненулевое значение из одного столбца, а также значение из другого столбца в той же строке в Polars?

Но, к сожалению, это не решает исходную проблему, поскольку вопрос выполняет агрегацию всего столбца, а затем принимает последнее значение этого столбца.
Что я хотел бы сделать заключается не в агрегировании всего столбца, а просто в вычитании текущего значения из предыдущего ненулевого значения.
Я также пытался использовать прокрутку:

Код: Выделить всё

df = df.with_row_index().rolling(
index_column = 'index',
period = '???i').agg(pl.col("A").last())

Но, конечно, это не работает, поскольку возникновение нулевых значений невозможно определить (т. е. оно не является периодическим, поэтому я не знаю, сколько индексов до того, как текущая запись содержит ненулевое значение в "A").
Кто-нибудь знает, как это сделать?
Спасибо!>

Подробнее здесь: https://stackoverflow.com/questions/792 ... -in-polars