Início / coding / Perguntas / 79561147
Accepted
GH KIM
Asked: 2025-04-08 12:01:48 +0800 CST

Alterar tipo de coluna no Polars DataFrame

  • 772

Tenho um Polars DataFrame abaixo.

import polars as pl
df = pl.DataFrame({"a":["1.2", "2.3", "5.4"], 
                   "b":["0.4", "0.03", "0.12"],
                   "c":["AA", "BB", "CC"]})

>>> df
     a        b          c
    str      str       str
  -------------------------
  "1.2"    "0.04"      "AA"
  "2.3"    "0.3"       "BB"
  "3.5"    "0.12"      "CC"

Como posso converter as colunas para tipos específicos? Neste caso, quero converter colunas ae bem floats.

Espero abaixo.

>>> df
    a        b          c
   f64      f64       str
  ------------------------
  1.2      0.04      "AA"
  2.3      0.3       "BB"
  3.5      0.12      "CC"
python

1 respostas

  1. Best Answer

    Existem várias abordagens no Polars para converter o tipo de coluna. Usar o castmétodo with with_columnsé uma das abordagens mais eficientes:

    df = df.with_columns([
    pl.col("a").cast(pl.Float64),
    pl.col("b").cast(pl.Float64)
])

    Outra abordagem:

    df = df.select([
    pl.col("a").cast(pl.Float64),
    pl.col("b").cast(pl.Float64),
    pl.col("c")
])

    E:

    df = df.cast({"a": pl.Float64, "b": pl.Float64})

    Para conjuntos de dados grandes, você pode tentar:

    df = df.lazy().with_columns([
    pl.col("a").cast(pl.Float64),
    pl.col("b").cast(pl.Float64)
])
df_result = df.collect()

    Para mais informações: https://docs.pola.rs/user-guide/expressions/casting/

    Saída:

    ┌─────┬──────┬─────┐
│ a   ┆ b    ┆ c   │
│ --- ┆ ---  ┆ --- │
│ f64 ┆ f64  ┆ str │
╞═════╪══════╪═════╡
│ 1.2 ┆ 0.4  ┆ AA  │
│ 2.3 ┆ 0.03 ┆ BB  │
│ 5.4 ┆ 0.12 ┆ CC  │
└─────┴──────┴─────┘
    • 3

relate perguntas