test tutte le combinazioni in Python

Question

Ho due gruppi di opzioni:

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

Il numero degli elementi nelle optionlists non sono necessariamente uguali e devo scegliere tra la prima optionlist due volte. Come posso assicurarmi di aver provato ogni combinazione di 2 opzioni dalla prima lista e una dalla seconda lista. Un esempio di selezione impostato sotto ...

selectedOptions = [an1,an2,bn] 

Answer 1

Supponendo che non si desidera voci duplicate da lista1, ecco un generatore che è possibile utilizzare per iterare su tutte le combinazioni:

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

Questo, tuttavia, non seleziona le stesse opzioni da lista1 in diversi ordini. Se si vuole ottenere sia [a1, a2, b1] e [a2, a1, b1] allora si può usare:

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

Un modo per fare è utilizzare itertools.product.

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

Answer 3

È possibile utilizzare itertools.product per questo. Restituisce tutte le possibili combinazioni.

Ad esempio

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

Ciò produrrà "doppi" come [a1, a1, b2] e [a2, a3, b2], [a3, a2, b2]. Puoi sistemarlo con un filtro. Di seguito le impedisce eventuali doppie *:

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

(*) Questo presuppone che le opzioni hanno qualche ordinamento naturale che sarà il caso di tutti i valori di base.

shang 's answer è anche molto buono. Ho scritto un codice per confrontarli:

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 

e il risultato è:

0.189330005646 
0.428138256073 

Quindi il metodo generatore è più veloce. Tuttavia, la velocità non è tutto. Personalmente trovo il mio codice 'più pulito', ma la scelta è tua!

(Btw, danno entrambi i fronti identici, quindi entrambi sono ugualmente corrette.)

Answer 4

Sembra a me come siete alla ricerca di 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

Unire product e permutations da itertools supponendo non voglio duplicati dal primo elenco:

>>> 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