typestar

Agrupar y agregar en Python

Resumir grupos con expresiones de agregación.

import polars as pl

df = pl.DataFrame({
    "team": ["a", "b", "a", "b", "a"],
    "player": ["kim", "sam", "rio", "noor", "tui"],
    "points": [10, 7, 12, 9, 5],
})

por_equipo = df.group_by("team").agg(
    pl.col("points").sum().alias("total"),
    pl.col("points").mean().alias("avg"),
    pl.len().alias("players"),
)

clasificado = por_equipo.sort("total", descending=True)

Cómo funciona

  1. group_by('team') forma los grupos.
  2. agg calcula sumas, medias y pl.len por grupo.
  3. El resultado se ordena por el agregado.

Palabras clave y builtins usados aquí

El intento, en números

Líneas
15
Caracteres a escribir
357
Tokens
162
Ritmo de tres estrellas
110 tpm

Al ritmo de tres estrellas de 110 tokens por minuto, este intento toma unos 88 segundos.

Escribe este fragmento

Paso 1 de 3 en Agregación; paso 12 de 32 en Polars.

← Anterior Siguiente →