Agregar columnas en Python

with_columns evalúa expresiones contra el dataframe y devuelve uno nuevo.

import polars as pl

tabla = pl.DataFrame({"chars": [420, 380], "seconds": [60.0, 45.0]})

ancha = tabla.with_columns(
    (pl.col("chars") / pl.col("seconds")).alias("cps"),
    (pl.col("chars") / 5 / (pl.col("seconds") / 60)).round(1).alias("wpm"),
).with_columns(
    (pl.col("wpm") > 80).alias("fast"),
)

print(ancha)
print(tabla.columns)

Cómo funciona

Varias expresiones en una llamada se calculan en paralelo.
alias nombra el resultado; sin él se conserva el nombre de entrada.
Nada muta: el dataframe original queda intacto.

Palabras clave y builtins usados aquí

as
print

El intento, en números

Líneas: 13
Caracteres a escribir: 331
Tokens: 138
Ritmo de tres estrellas: 105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 79 segundos.

Escribe este fragmento

Paso 3 de 4 en Expresiones; paso 7 de 32 en Polars.

← Anterior Siguiente →