Как преобразовать строки в столбцы с произвольным размером шага в полярах (за один шаг)? ⇐ Python

[Anonymous]

Мне удалось решить эту проблему в два этапа.

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

import polars as pl
text = "a brown fox jumps over a lazy dog's head"
step = 3
df = pl.DataFrame({"a":text.split(" ")})

first =  df.filter(pl.int_range(pl.len())%step==0)
second = df.filter(pl.int_range(pl.len())%step==1)
third=   df.filter(pl.int_range(pl.len())%step==2)

dff = (
pl.DataFrame({
'first':first['a'],
'second':second['a'],
'third':third['a']})
)
print(dff)

shape: (3, 3)
┌───────┬────────┬───────┐
│ first ┆ second ┆ third │
│ ---   ┆ ---    ┆ ---   │
│ str   ┆ str    ┆ str   │
╞═══════╪════════╪═══════╡
│ a     ┆ brown  ┆ fox   │
│ jumps ┆ over   ┆ a     │
│ lazy  ┆ dog's  ┆ head  │
└───────┴────────┴───────┘
#

У меня сложилось впечатление, что это можно легко решить с помощью одной цепочки выражений, но мне это не удалось. Есть предложения?


Подробнее здесь: https://stackoverflow.com/questions/735 ... rs-in-a-si