Agrupar elementos por clave en Python

El GROUP BY de SQL, en tres líneas de Python.

from collections import defaultdict

def agrupar_por(elementos, clave):
    grupos = defaultdict(list)
    for elemento in elementos:
        grupos[clave(elemento)].append(elemento)
    return dict(grupos)

Cómo funciona

1. defaultdict(list) crea la cubeta la primera vez que aparece una clave.
2. Cada elemento cae en la cubeta que elige su función de clave.
3. Devuelve los grupos como un dict común.

Palabras clave y builtins usados aquí

• def
• dict
• for
• list
• return

El intento, en números

Líneas

7

Caracteres a escribir

186

Tokens

40

Ritmo de tres estrellas

100 tpm

Al ritmo de tres estrellas de 100 tokens por minuto, este intento toma unos 24 segundos.

Escribe este fragmento

Paso 10 de 10 en Colecciones; paso 32 de 72 en Fundamentos del lenguaje.

← Anterior Siguiente →