Получить список имен столбцов со значениями > 0 для каждой строки в полярах. ⇐ Python

[Anonymous]

Я хочу добавить результат столбца в DataFrame Polars, который содержит список имен столбцов со значением больше нуля в этой позиции.
Итак, учитывая это:
import polars as pl

df = pl.DataFrame({"apple": [1, 0, 2, 0], "banana": [1, 0, 0, 1]})
cols = ["apple", "banana"]

shape: (4, 2)
┌───────┬────────┐
│ apple ┆ banana │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═══════╪════════╡
│ 1 ┆ 1 │
│ 0 ┆ 0 │
│ 2 ┆ 0 │
│ 0 ┆ 1 │
└───────┴────────┘

Как мне получить:
shape: (4, 3)
┌───────┬────────┬─────────────────────┐
│ apple ┆ banana ┆ result │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ list[str] │
╞═══════╪════════╪═════════════════════╡
│ 1 ┆ 1 ┆ ["apple", "banana"] │
│ 0 ┆ 0 ┆ [] │
│ 2 ┆ 0 ┆ ["apple"] │
│ 0 ┆ 1 ┆ ["banana"] │
└───────┴────────┴─────────────────────┘

Все, что у меня пока есть, — это истинностные значения:
df.with_columns(pl.concat_list(pl.col(cols).gt(0)).alias("result"))

shape: (4, 3)
┌───────┬────────┬────────────────┐
│ apple ┆ banana ┆ result │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ list[bool] │
╞═══════╪════════╪════════════════╡
│ 1 ┆ 1 ┆ [true, true] │
│ 0 ┆ 0 ┆ [false, false] │
│ 2 ┆ 0 ┆ [true, false] │
│ 0 ┆ 1 ┆ [false, true] │
└───────┴────────┴────────────────┘


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