Elegir columnas en Python

select toma expresiones: un conjunto de columnas se describe en vez de enumerarse.

import polars as pl

tabla = pl.DataFrame({
    "lang": ["python", "rust"],
    "tours": [11, 12],
    "steps": [299, 181],
    "tpm": [104.5, 98.0],
})

print(tabla.select(pl.col("lang", "tpm")))
print(tabla.select(pl.all().exclude("lang")).columns)
print(tabla.select(pl.col(pl.Int64)).columns)
print(tabla.select(pl.all().exclude(pl.Utf8)).columns)

Cómo funciona

pl.all() es cada columna; exclude quita de un conjunto.
pl.col acepta varios nombres, o un dtype.
Los selectores se componen, y eso vuelve manejables los dataframes anchos.

Palabras clave y builtins usados aquí

as
print

El intento, en números

Líneas: 13
Caracteres a escribir: 335
Tokens: 139
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 2 de 4 en Expresiones; paso 6 de 32 en Polars.

← Anterior Siguiente →