groupby en Python

Dividir por una clave, aplicar una función, combinar las respuestas.

import pandas as pd

tabla = pd.DataFrame({
    "lang": ["python", "python", "rust", "rust", "sql"],
    "stars": [3, 1, 3, 2, 2],
    "tpm": [104, 91, 98, 88, 79],
})

print(tabla.groupby("lang")["tpm"].mean().round(1))
print(tabla.groupby("lang").size())
print(tabla.groupby("lang", as_index=False)["stars"].max())

Cómo funciona

1. Un solo agregado sobre un groupby da una fila por grupo.
2. as_index=False deja la clave como columna en vez de índice.
3. size cuenta filas; nunique cuenta valores distintos.

Palabras clave y builtins usados aquí

• as
• print

El intento, en números

Líneas

11

Caracteres a escribir

304

Tokens

135

Ritmo de tres estrellas

110 tpm

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

Escribe este fragmento

Paso 1 de 5 en Agrupación; paso 14 de 26 en pandas.

← Anterior Siguiente →