Início / coding / Perguntas / 78988010
Accepted
DJDuque
Asked: 2024-09-16 01:45:39 +0800 CST

Explodir várias colunas com comprimentos diferentes

  • 772

Eu tenho um dataframe como:

data = {
    "a": [[1], [2], [3, 4], [5, 6, 7]],
    "b": [[], [8], [9, 10], [11, 12]],
}
df = pl.DataFrame(data)
"""
┌───────────┬───────────┐
│ a         ┆ b         │
│ ---       ┆ ---       │
│ list[i64] ┆ list[i64] │
╞═══════════╪═══════════╡
│ [1]       ┆ []        │
│ [2]       ┆ [8]       │
│ [3, 4]    ┆ [9, 10]   │
│ [5, 6, 7] ┆ [11, 12]  │
└───────────┴───────────┘
"""

Cada par de listas pode não ter o mesmo comprimento, e eu quero "truncar" a explosão para a menor das duas listas:

"""
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 2   ┆ 8   │
│ 3   ┆ 9   │
│ 4   ┆ 10  │
│ 5   ┆ 11  │
│ 6   ┆ 12  │
└─────┴─────┘
"""

Eu estava pensando que talvez eu tivesse que preencher o menor de ambas as listas com Nonepara combinar ambos os comprimentos, e então drop_nulls. Mas eu estava pensando se havia uma abordagem mais direta para isso?

python

2 respostas

  1. Best Answer

    Aqui está uma abordagem:

    min_length = pl.min_horizontal(pl.col('a', 'b').list.len())

out = (df.filter(min_length != 0)
       .with_columns(
           pl.col('a', 'b').list.head(min_length)
           )
       .explode('a', 'b')
       )

    Saída:

    shape: (5, 2)
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 2   ┆ 8   │
│ 3   ┆ 9   │
│ 4   ┆ 10  │
│ 5   ┆ 11  │
│ 6   ┆ 12  │
└─────┴─────┘

    Explicação

    • 3

  2. Não consigo colocar nos comentários, então vou colocar outra resposta aqui, pois é um pouco mais genérica para o exemplo dado:

    (
    df.with_columns(
        pl.all().list.head(
            pl.min_horizontal(pl.all().list.len())
        )
    ).explode("*")
    .drop_nulls()
)

    shape: (5, 2)
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 2   ┆ 8   │
│ 3   ┆ 9   │
│ 4   ┆ 10  │
│ 5   ┆ 11  │
│ 6   ┆ 12  │
└─────┴─────┘
    • 0

relate perguntas