Início / coding / Perguntas / 78856001
Accepted
Dante
Asked: 2024-08-10 20:24:13 +0800 CST

Como combinar duas colunas em pares `{key:value}` em polares?

  • 772

Estou trabalhando com a Polars DataFramee quero combinar duas colunas em um formato de dicionário, onde os valores de uma coluna se tornam as chaves e os valores da outra coluna se tornam os valores correspondentes.

Aqui está um exemplo de DataFrame:

import polars as pl

df = pl.DataFrame({
    "name": ["Chuck", "John", "Alice"],
    "surname": ["Dalliston", "Doe", "Smith"]
})

Quero transformar esse DataFrame em uma nova coluna que contenha dicionários, onde nome é a chave e sobrenome é o valor. O resultado esperado deve ser assim:

shape: (3, 3)
┌───────┬─────────┬──────────────────────────┐
│ name  │ surname │ name_surname             │
│ ---   │ ---     │ ---                      │
│ str   │ str     │ dict[str, str]           │
├───────┼─────────┼──────────────────────────┤
│ Chuck │ Dalliston│ {"Chuck": "Dalliston"}   │
│ John  │ Doe     │ {"John": "Doe"}          │
│ Alice │ Smith   │ {"Alice": "Smith"}       │
└───────┴─────────┴──────────────────────────┘

Eu tentei o seguinte código:

df.with_columns(
    json = pl.struct("name", "surname").map_elements(json.dumps)
)

Mas o resultado não é o esperado. Em vez de criar um dicionário com key-value, produz:

{name:Chuck,surname:Dalliston}
python

2 respostas

  1. Best Answer

    Você pode tentar este trecho de código. Este parece ser o mais próximo que você pode chegar, pois não possui um ditado ingênuo.

    Veja referência: data_types_polaris

    import polars as pl

df = pl.DataFrame(
    {"name": ["Chuck", "John", "Alice"], "surname": ["Dalliston", "Doe", "Smith"]}
)

df = df.select(
    [
        "name",
        "surname",
        (
            pl.struct(["name", "surname"]).map_elements(
                lambda row: {row["name"]: row["surname"]}, return_dtype=pl.Object
            )
        ).alias("name_surname"),
    ]
)
print(df)

    ┌───────┬───────────┬────────────────────────┐
│ name  ┆ surname   ┆ name_surname           │
│ ---   ┆ ---       ┆ ---                    │
│ str   ┆ str       ┆ object                 │
╞═══════╪═══════════╪════════════════════════╡
│ Chuck ┆ Dalliston ┆ {'Chuck': 'Dalliston'} │
│ John  ┆ Doe       ┆ {'John': 'Doe'}        │
│ Alice ┆ Smith     ┆ {'Alice': 'Smith'}     │
└───────┴───────────┴────────────────────────┘
    • 1 
  2. import polars as pl

df = pl.DataFrame(
    {"name": ["Chuck", "John", "Alice"], "surname": ["Dalliston", "Doe", "Smith"]}
)
df = df.with_columns(
            json=pl.concat_str(
                pl.lit("{"),
                pl.col("specification_name"),
                pl.lit(":"),
                pl.col("specification_value"),
                pl.lit("}"),
                separator='"',
            )
        )

    Isso retornará um json como string

    • 0

relate perguntas