So gruppieren Sie Datenrahmenzeilen in Polars group_by in einer Liste ⇐ Python

[Anonymous]

Code: Select all

import polars as pl

df = pl.DataFrame({
"Letter": ["A", "A", "B", "B", "B", "C", "C", "D", "D", "E"],
"Value': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
})

Ich möchte Buchstaben gruppieren und ihren entsprechenden Wert in einer Liste sammeln.
Verwandte Pandas-Frage: So gruppieren Sie Datenrahmenzeilen in einer Liste in Pandas Groupby
Ich weiß, dass Pandas-Code hier nicht funktioniert:

Code: Select all

df.group_by("a")["b"].apply(list)

TypeError: „GroupBy“-Objekt ist nicht abonnierbar

Die Ausgabe lautet:

Code: Select all

┌────────┬───────────┐
│ Letter ┆ Value     │
│ ---    ┆ ---       │
│ str    ┆ list[i64] │
╞════════╪═══════════╡
│ A      ┆ [1, 2]    │
│ B      ┆ [3, 4, 5] │
│ C      ┆ [6, 7]    │
│ D      ┆ [8, 9]    │
│ E      ┆ [10]      │
└────────┴───────────┘