itertools

itertools proporciona bloques de construcción de iteradores rápidos y eficientes en memoria: encadenamiento, segmentación, combinatoria y agrupación sin listas intermedias.

Receta

from itertools import chain, islice, product
 
flat = chain([1, 2], [3, 4])
first_three = list(islice(range(10), 3))
pairs = list(product(["A", "B"], [1, 2]))

Cuándo usar esto:

• Aplanar iterables anidados
• Paginación a través de islice en un contador infinito
• Agrupar líneas de registro ordenadas por clave
• Productos cartesianos para la generación de casos de prueba

Ejemplo de Trabajo

from itertools import chain, groupby, islice, accumulate
 
def group_sorted(rows: list[tuple[str, int]]):
    for key, group in groupby(rows, key=lambda r: r[0]):
        yield key, sum(v for _, v in group)
 
def paginate(iterator, page_size: int):
    page = 0
    while True:
        chunk = list(islice(iterator, page_size))
        if not chunk:
            break
        yield page, chunk
        page += 1
 
def running_totals(values):
    return accumulate(values)
 
if __name__ == "__main__":
    rows = [("a", 1), ("a", 2), ("b", 3)]
    print(list(group_sorted(sorted(rows))))
    print(list(paginate(iter(range(7)), 3)))
    print(list(running_totals([1, 2, 3, 4])))

Lo que esto demuestra:

• groupby necesita ordenar primero por la misma función de clave
• islice toma de cualquier iterador sin len()
• Patrón paginate para APIs de streaming
• accumulate produce reducciones en ejecución (suma por defecto)

Inmersión Profunda

Cómo Funciona

• Álgebra de iteradores - Las funciones consumen iterables, devuelven iteradores.
• groupby - Agrupa solo claves consecutivas iguales, no es GROUP BY de SQL.
• tee - Divide el iterador en iteradores independientes con almacenamiento en búfer.
• cycle/repeat/count - Iteradores infinitos; el bucle consumidor siempre está limitado.
• combinations/permutations - Generación combinatoria perezosa.

Usados Frecuentemente

Notas de Python

from itertools import pairwise  # 3.10+
 
for a, b in pairwise([1, 2, 3]):
    print(a, b)

Trampas

• groupby sin ordenar - Divide los grupos incorrectamente cuando las claves no son contiguas. Solución: sorted(data, key=key) primero.
• Memoria de tee - Almacena valores en búfer para iteradores secundarios. Solución: Usar con moderación en flujos grandes.
• Producto infinito - product(count(), repeat=2) nunca termina. Solución: Limitar con islice.
• chain.from_iterable - Aplana un nivel; la anidación más profunda requiere recursión o chain anidado.
• zip vs zip_longest - zip se detiene en el más corto; zip_longest necesita fillvalue para el relleno.

Alternativas

Preguntas Frecuentes

Relacionado

• Generadores y yield - iteradores personalizados
• functools - reduce relacionado con accumulate
• Comprensiones y Expresiones Generadoras - genexp
• Módulo collections - Counter con iterables

Versiones de la pila: Esta página fue escrita para Python 3.14.0 (estable 3.14, mantenimiento 3.13), FastAPI 0.115+, Django 5.2, Flask 3.1, Pydantic 2, PyTorch 2.6+, pandas 2.2+, Polars 1.x, ruff 0.9+, y uv 0.6+.