Prueba todas las combinaciones en Python

Question

que tienen dos grupos de opciones:

optionList1 = [a1,a2,a3,...,an] 
optionList2 = [b1,b2,b3,...,bn] 

El número de elementos en los optionlists no son necesariamente iguales y tengo que elegir entre el primer listaOpciones dos veces. ¿Cómo me aseguro de haber probado todas las combinaciones de 2 opciones de la primera lista y una de la segunda lista? Una selección de ejemplo establecida a continuación ...

selectedOptions = [an1,an2,bn] 

Answer 1

Suponiendo que usted no desea que las entradas duplicadas de lista1, aquí es un generador que se puede utilizar para iterar sobre todas las combinaciones:

def combinations(list1, list2): 
    return ([opt1, opt2, opt3] 
      for i,opt1 in enumerate(list1) 
      for opt2 in list1[i+1:] 
      for opt3 in list2) 

Esto, sin embargo, no selecciona las mismas opciones de list1 en diferentes ordenamientos. Si desea obtener tanto [a1, a2, b1] y [A2, A1, B1] entonces usted puede utilizar:

def combinations(list1, list2): 
    return ([opt1, opt2, opt3] 
      for opt1 in list1 
      for opt2 in list1 
      for opt3 in list2 if opt1 != opt2) 

Answer 2

Una forma de hacerlo es usar itertools.product.

for x, y, z in itertools.product(optionlist1,optionlist1,optionlist2): 
    print x,y,z 

Answer 3

Puede utilizar itertools.product para esto. Devuelve todas las combinaciones posibles.

Por ejemplo

for a1, a2, b in itertools.product(optionlist1,optionlist1,optionlist2): 
    do_something(a1,a2,b) 

Esto producirá "dobles" como [a1, a1, b2] y [A2, A3, B2], [a3, a2, b2]. Puedes arreglar esto con un filtro. A continuación se previene cualquier dobles *:

for a1,a2,b in itertools.ifilter(lambda x: x[0]<x[1], itertools.product(optionlist1,optionlist1,optionlist2)): 
    do_something(a1,a2,b) 

(*) Esto asume que las opciones tienen algún ordenamiento natural que será el caso con todos los valores primitivos.

shang 's answer también es muy bueno. Escribí algo de código para compararlos:

from itertools import ifilter, product 
import random 
from timeit import repeat 

def generator_way(list1, list2): 
    def combinations(list1, list2): 
     return ([opt1, opt2, opt3] 
       for i,opt1 in enumerate(list1) 
       for opt2 in list1[i+1:] 
       for opt3 in list2) 
    count = 0 
    for a1,a2,b in combinations(list1,list2): 
     count += 1 

    return count 

def itertools_way(list1,list2): 
    count = 0 
    for a1,a2,b in ifilter(lambda x: x[0] < x[1], product(list1,list1,list2)): 
     count += 1 
    return count 

list1 = range(0,100) 
random.shuffle(list1) 
list2 = range(0,100) 
random.shuffle(list2) 

print sum(repeat(lambda: generator_way(list1,list2),repeat = 10, number=1))/10 
print sum(repeat(lambda: itertools_way(list1,list2),repeat = 10, number=1))/10 

y el resultado es:

0.189330005646 
0.428138256073 

lo tanto, el método de generador es más rápido. Sin embargo, la velocidad no es todo. Personalmente encuentro mi código 'más limpio', ¡pero la elección es tuya!

(Por cierto, dan ambos casos idénticos, por lo que ambos son igualmente correctas.)

Answer 4

Me suena como que está buscando itertools.product()

>>> options_a = [1,2] 
>>> options_b = ['a','b','c'] 
>>> list(itertools.product(options_a, options_a, options_b)) 
[(1, 1, 'a'), 
(1, 1, 'b'), 
(1, 1, 'c'), 
(1, 2, 'a'), 
(1, 2, 'b'), 
(1, 2, 'c'), 
(2, 1, 'a'), 
(2, 1, 'b'), 
(2, 1, 'c'), 
(2, 2, 'a'), 
(2, 2, 'b'), 
(2, 2, 'c')] 

Answer 5

Combinar product y permutations de itertools suponiendo que no quiero duplicados de la primera lista:

>>> from itertools import product,permutations 
>>> o1 = 'a1 a2 a3'.split() 
>>> o2 = 'b1 b2 b3'.split() 
>>> for (a,b),c in product(permutations(o1,2),o2): 
...  print a,b,c 
... 
a1 a2 b1 
a1 a2 b2 
a1 a2 b3 
a1 a3 b1 
a1 a3 b2 
a1 a3 b3 
a2 a1 b1 
a2 a1 b2 
a2 a1 b3 
a2 a3 b1 
a2 a3 b2 
a2 a3 b3 
a3 a1 b1 
a3 a1 b2 
a3 a1 b3 
a3 a2 b1 
a3 a2 b2 
a3 a2 b3