Jak uzyskać wszystkie możliwe kombinacje elementów z 2-wymiarowej listy w python?

Question

Nie znalazłem lepszego sposobu na wyrażenie tego pytania w tytule. Jeśli możesz, edytuj.

Mam listę list tak:

a = [['a','b'],[1,2]] 

teraz chciałbym funkcję, która wyrzuca z siebie wszelką możliwą kombinację takiego:

[['a',1],['a',2],['b',1],['b',2]] 

gdzie ani liczby list in a jest znane z wyprzedzeniem, ani też nie jest znana długość każdej z podanych list podrzędnych, ale wszystkie kombinacje, które wyjdą, powinny zawierać 1 element z każdej podlisty.

Answer 1

Trzeba itertools.product():

>>> list(itertools.product(*a)) 
[('a', 1), ('a', 2), ('b', 1), ('b', 2)] 

Answer 2

To może być co itertools.product() (który wspomina Sven) robi:

def combs(list1, list2): 
    results = [] 
    for x in list1: 
     for y in list2: 
      l.append([x,y]) 
    return results 

Answer 3

Oto rozwiązanie przy użyciu rekursji, combs_r ma accum strawienia head (następna lista w linii), aby wytworzyć grubszy accum0, następnie wywołuje się ("rekursja") z tail (pozostałe listy) i teraz grubszym akumulatorem accum0.

Może być ciężkim użytkownikiem pamięci, ponieważ każde wywołanie do combs_r dodaje nową przestrzeń nazw, aż do końca, gdy wszystko się rozwinie. Osoba bardziej kompetentna w interpreterze Pythona może komentować to.

Płaci, aby dowiedzieć się prolog, IMHO.

def combs(ll): 
    if len(ll) == 0: 
     return [] 
    if len(ll) == 1: 
     return [[item] for item in ll[0]] 
    elif len(ll) == 2: 
     return lmul(ll[0], [[item] for item in ll[1]]) 
    else: 
     return combs_r(ll[1:], ll[0]) 

def combs_r(ll, accum): 
    head = ll[0] 
    tail = ll[1:] 
    accum0 = [] 
    accum0 = lmul(head, accum) 
    if len(tail) == 0: 
     return accum0 
    else: 
     return combs_r(tail, accum0) 

def lmul(head, accum): 
    accum0 = [] 
    for ah in head: 
     for cc in accum: 
      #cc will be reused for each ah, so make a clone to mutate 
      cc0 = [x for x in cc] 
      cc0.append(ah) 
      accum0.append(cc0) 
    return accum0 

sampleip = [['a','b','c'],[1,2], ['A', 'B']] 
sampleip2 = [['a','b','c'],[1,2]] 
sampleip1 = [['a','b','c']] 
sampleip0 = [] 
print combs(sampleip0) 
print combs(sampleip1) 
print combs(sampleip2) 
print combs(sampleip)