Início / coding / Perguntas / 78102296
Accepted
Palestine
Asked: 2024-03-04 23:17:10 +0800 CST

Como agrupar linhas do dataframe em uma lista em polars group_by

  • 772
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]
    }
)

Quero agrupar Lettere coletar seus correspondentes Valueem uma lista.

Pergunta relacionada ao Pandas: Como agrupar linhas do dataframe em uma lista no pandas groupby

Eu sei que o código do pandas não funcionará aqui:

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

TypeError: 'GroupBy' object is not subscriptable

A saída será:

| A      ┆ [1, 2]    │
│ B      ┆ [3, 4, 5] │
│ C      ┆ [6, 7]    │
│ D      ┆ [8, 9]    │
│ E      ┆ [10]      |
python

2 respostas

  1. Muito simples: se você não especificar uma função de agregação em uma pl.DataFrame.group_by().agg()construção, todos os valores de um grupo serão agregados em uma lista.

    df.group_by("Letter").agg("Value")

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

    Você poderia fazer isso. maintain_order=Trueé necessário se você quiser que a ordem dos grupos seja consistente com os dados de entrada .

    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]
    }
)
g = df.group_by('Letter', maintain_order=True).agg(pl.col('Value'))
print(g)

    Isto irá imprimir

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

relate perguntas