Как объединить head(n) и Tail(n) в одну группу_by с полярами ⇐ Python

[Anonymous]

Вот пример фрейма данных: я хочу получать строки head(n) и Tail(n) каждый день, что представляет собой group_by("date").agg() с полярами. Я знаю, что можно использовать два group_by, чтобы получить головную и хвостовую часть, а затем объединить их.

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

df = pl.from_repr("""
┌────────────┬────────┬──────────────────┐
│ date       ┆ symbol ┆ ts_dom2secdom_op │
│ ---        ┆ ---    ┆ ---              │
│ date       ┆ str    ┆ f64              │
╞════════════╪════════╪══════════════════╡
│ 2000-01-04 ┆ AL     ┆ -0.119165        │
│ 2000-01-04 ┆ RU     ┆ 0.256691         │
│ 2000-01-05 ┆ AL     ┆ -0.126549        │
│ 2000-01-05 ┆ RU     ┆ 0.1851           │
│ 2000-01-06 ┆ CU     ┆ -0.121354        │
│ 2000-01-06 ┆ RU     ┆ 0.228452         │
│ 2000-01-07 ┆ AL     ┆ -0.126013        │
│ 2000-01-07 ┆ RU     ┆ 0.348729         │
│ 2000-01-10 ┆ AL     ┆ -0.139447        │
│ 2000-01-10 ┆ RU     ┆ 0.263048         │
└────────────┴────────┴──────────────────┘
""")

Есть ли какой-нибудь трюк, чтобы сделать это в одном group_by, with_columns или фильтре?


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