Todas las variantes posibles de cremallera en Python

Question

Por ejemplo, tengo un código es el siguiente:

a = [1, 2] 
b = [4, 5] 

¿Cómo puedo obtener algo como esto:

[(1,4), (1,5), (2,4), (2,5)] 

Al igual que la función zip hace, pero con todas las variantes posibles O no puedo?

Answer 1

¿Quieres itertools.product:

>>> import itertools 
>>> a = [1,2] 
>>> b = [4,5] 
>>> list(itertools.product(a,b)) 
[(1, 4), (1, 5), (2, 4), (2, 5)] 

Answer 2

Si usted está interesado únicamente en el resultado, entonces itertools.product es lo que necesita (1 a @DSM para esto). Sin embargo, si está interesado en el algoritmo que genera algo como esto, se llama recursive descent. El algoritmo, en este caso, podría ser la siguiente (que sólo voy a imprimir los resultados aquí para mayor claridad):

def product(L, tmp=None): 
    if tmp is None: 
     tmp = [] 
    if L==[]: 
     print tmp 
    else: 
     for i in L[0]: 
      product(L[1:], tmp+[i]) 

Por lo tanto,

>>> product([[1,2], [4,5]]) 
[1, 4] 
[1, 5] 
[2, 4] 
[2, 5] 

Esperanza esto ayuda

Answer 3

Don 't lo obvio:

out = [] 
for a in [1, 2]: 
    for b in [4, 5]: 
     out.append((a, b)) 

o listas por comprensión:

a = [1, 2] 
b = [4, 5] 
out = [(x, y) for x in a for y in b] 

Ambos producen out == [(1, 4), (1, 5), (2, 4), (2, 5)]

Answer 4

Esto se puede hacer muy bien con la lista de comprensión, o mejor aún con una expresión generador si sólo tiene que recorrer a través de las combinaciones.

Así es que está usando listas por comprensión:

a = [1, 2] 
b = [4, 5] 

[(i, j) for i in a for j in b] 

Y aquí con un generador de expresión:

for pair in ((i, j) for i in a for j in b): 
    print(pair)