libreria library how examples combination python generator combinatorics

python - library - La forma pitónica de generar parejas.



python product() (6)

Como dijo @Sven, su código está intentando obtener todos los pares ordenados de elementos de las listas b . En este caso itertools.product(a,b) es lo que quieres. Si, por el contrario, realmente desea "combinaciones", que son pares desordenados de elementos distintos de la lista a , entonces desea itertools.combinations(a,2) .

>>> for pair in itertools.combinations([1,2,3,4],2): ... print pair ... (1, 2) (1, 3) (1, 4) (2, 3) (2, 4) (3, 4)

Quiero algo como el código a continuación, pero el estilo "pythonic" o el uso de la biblioteca estándar:

def combinations(a,b): for i in a: for j in b: yield(i,j)


Crear conjunto de pares (par, impar) combinación

>>> a = { (i,j) for i in range(0,10,2) for j in range(1,10,2)} >>> a {(4, 7), (6, 9), (0, 7), (2, 1), (8, 9), (0, 3), (2, 5), (8, 5), (4, 9), (6, 7), (2, 9), (8, 1), (6, 3), (4, 1), (4, 5), (0, 5), (2, 3), (8, 7), (6, 5), (0, 1), (2, 7), (8, 3), (6, 1), (4, 3), (0, 9)} def combinations(lista, listb): return { (i,j) for i in lista for j in listb } >>> combinations([1,3,5,6],[11,21,133,134,443]) {(1, 21), (5, 133), (5, 11), (5, 134), (6, 11), (6, 134), (1, 443), (3, 11), (6, 21), (3, 21), (1, 133), (1, 134), (5, 21), (3, 134), (5, 443), (6, 443), (1, 11), (3, 443), (6, 133), (3, 133)}


Estas no son realmente "combinaciones" en el sentido de combinatoria, son más bien elementos del producto cartesiano de a y b . La función en la biblioteca estándar para generar estos pares es itertools.product() :

for i, j in itertools.product(a, b): # whatever


La librería itertools tiene funciones combinatorias. Como dijo Sven, itertools.product sería la función apropiada en este caso:

list(itertools.product(''ab'', ''cd'')) [(''a'', ''c''), (''a'', ''d''), (''b'', ''c''), (''b'', ''d'')]


Una expresión de generador anidado también funcionará:

product = ((i, j) for i in a for j in b) for i, j in product: # ...


>>>a=[1,2,3] >>>b=[4,5,6] >>>zip(a,b) [(1, 4), (2, 5), (3, 6)]