Enumere con itertools

itertools de Python genera combinaciones y ordenaciones por usted, de modo que escribe menos código de bucles y comete menos errores. 🙂

Todo se encuentra en un único módulo, así que empiece con import itertools. Después puede llamar directamente a product, permutations y combinations.

Use product para reemplazar los bucles profundamente anidados sobre las mismas opciones. Genera cada tupla de selecciones a partir de los rangos indicados.

Cuando el orden importa, permutations enumera todas las disposiciones de los elementos. Es perfecto para una solución de fuerza bruta pequeña al estilo del problema del viajante.

Cuando el orden no importa, combinations selecciona cada grupo de tamaño r. Es ideal para elegir un subconjunto de elementos que comprobar.

La diferencia clave es el orden: permutations trata (1,2) y (2,1) como elementos distintos, mientras que combinations los considera la misma selección.

Las permutaciones crecen como n!, lo que aumenta rápidamente. Incluso n igual a 11 ya supera los cuarenta millones de disposiciones, así que mantenga n pequeño.

Estas funciones son generadores: producen un elemento a la vez sin construir una lista enorme, por lo que el consumo de memoria se mantiene bajo durante el bucle.

Genere las candidatas y, después, conserve solo las que superen su prueba. Este patrón convierte una búsqueda difícil en un bucle breve y legible.

¿Necesita elegir elementos que puedan repetirse? Use combinations_with_replacement para permitir que el mismo elemento aparezca más de una vez en un grupo.

Hágase dos preguntas: ¿importa el orden?, y ¿pueden repetirse los elementos? Las respuestas le indican directamente la función correcta de itertools.

Debe enumerar todas las formas de elegir 3 de 6 elementos cuando el orden no importa.

Use product, permutations y combinations para enumerar de forma clara. Elija la función preguntándose si importa el orden y si se repiten los elementos, y vigile el crecimiento. 🚀