Условное агрегирование строк в Polars на основе последовательных значений ⇐ Python

[Anonymous]

У меня есть данные с двумя столбцами в следующем формате:

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

import polars as pl

df = pl.from_repr("""
┌──────┬─────────┐
│ nums ┆ letters │
│ ---  ┆ ---     │
│ i64  ┆ str     │
╞══════╪═════════╡
│ 160  ┆ B       │
│ 540  ┆ I       │
│ 550  ┆ I       │
│ 590  ┆ B       │
│ 370  ┆ I       │
└──────┴─────────┘
""")

И я хочу объединить ячейки, имеющие форму B, в дополнение к последовательным I (т. е. BI*):

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

shape: (2, 1)
┌─────────────────┐
│ nums            │
│ ---             │
│ list[i64]       │
╞═════════════════╡
│ [160, 540, 550] │
│ [590, 370]      │
└─────────────────┘

Как я могу добиться чего-то подобного с помощью поляров?

Подробнее здесь: https://stackoverflow.com/questions/718 ... ive-values