Я хочу добавить результат столбца в Polars DataFrame, который содержит список имен столбцов со значением больше нуля в этой позиции.
Итак, учитывая это:
import polars as pl
df = pl.DataFrame({"apple": [1, 0, 2, 0], "banana": [1, 0, 0, 1]})
cols = ["apple", "banana"]
Как мне получить:
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
