Início / coding / Perguntas / 77354114
Accepted
Thomas
Asked: 2023-10-25 01:14:15 +0800 CST

Como eu geraria combinações de itens dentro do Polars usando a API de expressão nativa?

  • 772

Existe uma maneira de gerar combinações de itens dentro de uma lista dentro de uma coluna Polars sem recorrer ao .map_elements()+ itertools para cada linha?

Esta é a minha solução atual:

import polars as pl
import itertools

(pl.DataFrame({'col': [['a', 'b', 'c']]})
   .with_columns(pl.col('col')
                   .map_elements(lambda list_o_things: [sorted((thing_1, thing_2))
                                                        for thing_1, thing_2 
                                                        in itertools.combinations(list_o_things, 2)])
                )
)

que retorna isso:

[['a', 'b'], ['a', 'c'], ['b', 'c']]

list

1 respostas

  1. Best Answer

    Explodir a estrutura aninhada, fazer uma junção cruzada com ela mesma, filtrar as entradas redundantes, concatenar para a lista e implodir para a lista aninhada.

    df=pl.DataFrame({'col': [['a', 'b', 'c']]})
(
    df
    .explode('col')
    .join(
        df.explode('col'), how='cross')
    .filter(pl.col('col')<pl.col('col_right'))
    .select(pl.concat_list('col','col_right').implode())
    )
shape: (1, 1)
┌──────────────────────────────────────┐
│ col                                  │
│ ---                                  │
│ list[list[str]]                      │
╞══════════════════════════════════════╡
│ [["a", "b"], ["a", "c"], ["b", "c"]] │
└──────────────────────────────────────┘
    • 1

relate perguntas