Поэлементное агрегирование столбца типа List[f64] в Polars ⇐ Python

[Anonymous]

Я хочу применить функции агрегирования, такие как сумма, среднее значение и т. д., к столбцу типа List[f64] после group_by, чтобы получить обратно запись List[64].

Скажем, у меня есть:

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

import polars as pl

df = pl.DataFrame(
{
"Case": ["case1", "case1"],
"List": [[1, 2, 3], [4, 5, 6]],
}
)

print(df)

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

shape: (2, 2)
┌───────┬────────────┐
│ Case  ┆ List       │
│ ---   ┆ ---        │
│ str   ┆ list[i64]  │
╞═══════╪════════════╡
│ case1 ┆ [1, 2, 3]  │
│ case1 ┆ [4, 5, 6]  │
└───────┴────────────┘

Я хочу сгруппировать по регистру и суммировать список, чтобы в итоге получилось:

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

┌───────┬────────────┐
│ Case  ┆ List       │
│ ---   ┆ ---        │
│ str   ┆ list[i64]  │
╞═══════╪════════════╡
│ case1 ┆ [5, 7, 9]  │
└───────┴────────────┘

Как мне лучше всего это сделать? Обратите внимание, что длина каждого из списков равна 256, поэтому индексирование каждого из них не является хорошим решением.
Спасибо!

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