Início / coding / Perguntas / 78816286
Accepted
Phil-ZXX
Asked: 2024-07-31 20:18:14 +0800 CST

Crie um dataframe polar a partir de dict (com chaves e valores sendo suas próprias colunas)

  • 772

eu tenho o seguinte código

import polars as pl

mapping = {
    'CASH':  {'qty':  1, 'origin': 'E'},
    'ITEM':  {'qty': -9, 'origin': 'A'},
    'CHECK': {'qty': 46, 'origin': 'A'},
}

df = pl.DataFrame([{'type': k} | v for k, v in mapping.items()])\
         .with_columns(pl.struct(['qty', 'origin']).alias('mapping'))\
         .select(pl.col(['type', 'mapping']))

Portanto, as chaves do dict devem se tornar uma nova coluna chamada typee os valores do dict devem estar em sua própria mappingcoluna. Minha implementação acima funciona e dfse parece com isto:

shape: (3, 2)
┌───────┬───────────┐
│ type  ┆ mapping   │
│ ---   ┆ ---       │
│ str   ┆ struct[2] │
╞═══════╪═══════════╡
│ CASH  ┆ {1,"E"}   │
│ ITEM  ┆ {-9,"A"}  │
│ CHECK ┆ {46,"A"}  │
└───────┴───────────┘

Mas minha implementação é muito longa e não parece muito eficiente. Existe uma maneira polar mais idiomática de criar esse dataframe?

dataframe

2 respostas

  1. O construtor padrão DataFrameaceita dicionário. Basta especificar os nomes das colunas e usar dict.keys()e dict.values()como dados.

    pl.DataFrame({
    "type": mapping.keys(),
    "mapping": mapping.values()
})

┌───────┬───────────┐
│ type  ┆ mapping   │
│ ---   ┆ ---       │
│ str   ┆ struct[2] │
╞═══════╪═══════════╡
│ CASH  ┆ {1,"E"}   │
│ ITEM  ┆ {-9,"A"}  │
│ CHECK ┆ {46,"A"}  │
└───────┴───────────┘
    • 1
  2. Best Answer

    Você pode criar as 2 colunas diretamente, conforme especificado:

    df = pl.DataFrame({
    "type": mapping.keys(), 
    "mapping": mapping.values()
})

    shape: (3, 2)
┌───────┬───────────┐
│ type  ┆ mapping   │
│ ---   ┆ ---       │
│ str   ┆ struct[2] │
╞═══════╪═══════════╡
│ CASH  ┆ {1,"E"}   │
│ ITEM  ┆ {-9,"A"}  │
│ CHECK ┆ {46,"A"}  │
└───────┴───────────┘
    • 1

relate perguntas