Wszystkie kombinacje listy list

240

Po prostu szukam pythonowej wersji Combination ofList<List<int>>

Biorąc pod uwagę listę list, potrzebuję nowej listy, która zawiera wszystkie możliwe kombinacje elementów między listami.

[[1,2,3],[4,5,6],[7,8,9,10]] -> [[1,4,7],[1,4,8],...,[3,6,10]]

Liczba list jest nieznana, więc potrzebuję czegoś, co zadziała we wszystkich przypadkach. Punkty bonusowe za elegancję!

Lin
źródło

Odpowiedzi:

428

potrzebujesz itertools.product:

>>> import itertools
>>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
>>> list(itertools.product(*a))
[(1, 4, 7), (1, 4, 8), (1, 4, 9), (1, 4, 10), (1, 5, 7), (1, 5, 8), (1, 5, 9), (1, 5, 10), (1, 6, 7), (1, 6, 8), (1, 6, 9), (1, 6, 10), (2, 4, 7), (2, 4, 8), (2, 4, 9), (2, 4, 10), (2, 5, 7), (2, 5, 8), (2, 5, 9), (2, 5, 10), (2, 6, 7), (2, 6, 8), (2, 6, 9), (2, 6, 10), (3, 4, 7), (3, 4, 8), (3, 4, 9), (3, 4, 10), (3, 5, 7), (3, 5, 8), (3, 5, 9), (3, 5, 10), (3, 6, 7), (3, 6, 8), (3, 6, 9), (3, 6, 10)]
SilentGhost
źródło
20
Czy ktoś mógłby wyjaśnić znaczenie gwiazdki *a?
Serrano
52
*aoznacza, że ​​są to argumenty przekazywane do funkcji lub metody. def fn(a,b,c):odpowiedziałby na fn(*[1,2,3]) odniesienie
mjallday
1
@mjallday, czy byłoby możliwe dodanie również tych kombinacji: (7,4,1), (8,4,1), (9,4,1), (10,4,1), (7,5, 1), (8,5,1), (9,5,1), (10,5,1) itd.?
Reman
1
@Reman Nie jest do końca jasne, co chcesz uzyskać, ale jeśli jest to na przykład także odwrotność każdej krotki, możesz użyć funkcji otoki, która przyjmuje ajako dane wejściowe, iteruje itertools.product(*a)i yieldjest zarówno krotką wyprodukowaną przez, jak itertoolsi wersją odwróconą ( np. utwórz listę reverse()i przekonwertuj ją z powrotem na krotkę). Najlepiej zadaj nowe pytanie.
Joachim Wagner
24

Najbardziej eleganckim rozwiązaniem jest użycie itertools.product w Pythonie 2.6.

Jeśli nie używasz języka Python 2.6, dokumenty dla itertools.product faktycznie pokazują równoważną funkcję umożliwiającą wykonanie produktu w „ręczny” sposób:

def product(*args, **kwds):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = map(tuple, args) * kwds.get('repeat', 1)
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)
Jarret Hardie
źródło
19
listOLists = [[1,2,3],[4,5,6],[7,8,9,10]]
for list in itertools.product(*listOLists):
  print list;

Mam nadzieję, że uważasz to za tak eleganckie, jak wtedy, gdy go spotkałem.

Matthew Flaschen
źródło
5
O co chodzi z tym średnikiem? :)
Paolo Bergantino
3
Siła nawyku. Uwielbiam sposób, w jaki Python pozwala umieścić jeden średnik, tylko po to, aby pomóc nam programiście C / Java. Ale to jasne; tak naprawdę nie jest terminatorem instrukcji, gdy wykonujesz coś takiego jak print („foo”) ;; co jest całkowicie legalne w C lub Javie (choć bezcelowe), ale zakazane w Pythonie.
Matthew Flaschen
5

Numpy może to zrobić:

 >>> import numpy
 >>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
 >>> [list(x) for x in numpy.array(numpy.meshgrid(*a)).T.reshape(-1,len(a))]
[[ 1, 4, 7], [1, 5, 7], [1, 6, 7], ....]
Diamantatos Paraskevas
źródło
Czy ktoś mógłby to wyjaśnić?
ashishv
5

Nie ma nic złego w prostej rekursji dla tego zadania, a jeśli potrzebujesz wersji, która działa z łańcuchami, może to pasować do twoich potrzeb:

combinations = []

def combine(terms, accum):
    last = (len(terms) == 1)
    n = len(terms[0])
    for i in range(n):
        item = accum + terms[0][i]
        if last:
            combinations.append(item)
        else:
            combine(terms[1:], item)


>>> a = [['ab','cd','ef'],['12','34','56']]
>>> combine(a, '')
>>> print(combinations)
['ab12', 'ab34', 'ab56', 'cd12', 'cd34', 'cd56', 'ef12', 'ef34', 'ef56']
duanev
źródło
3

Można do tego użyć podstawowego pythona. Kod potrzebuje funkcji do spłaszczenia list list:

def flatten(B):    # function needed for code below;
    A = []
    for i in B:
        if type(i) == list: A.extend(i)
        else: A.append(i)
    return A

Następnie można uruchomić:

L = [[1,2,3],[4,5,6],[7,8,9,10]]

outlist =[]; templist =[[]]
for sublist in L:
    outlist = templist; templist = [[]]
    for sitem in sublist:
        for oitem in outlist:
            newitem = [oitem]
            if newitem == [[]]: newitem = [sitem]
            else: newitem = [newitem[0], sitem]
            templist.append(flatten(newitem))

outlist = list(filter(lambda x: len(x)==len(L), templist))  # remove some partial lists that also creep in;
print(outlist)

Wynik:

[[1, 4, 7], [2, 4, 7], [3, 4, 7], 
[1, 5, 7], [2, 5, 7], [3, 5, 7], 
[1, 6, 7], [2, 6, 7], [3, 6, 7], 
[1, 4, 8], [2, 4, 8], [3, 4, 8], 
[1, 5, 8], [2, 5, 8], [3, 5, 8], 
[1, 6, 8], [2, 6, 8], [3, 6, 8], 
[1, 4, 9], [2, 4, 9], [3, 4, 9], 
[1, 5, 9], [2, 5, 9], [3, 5, 9], 
[1, 6, 9], [2, 6, 9], [3, 6, 9], 
[1, 4, 10], [2, 4, 10], [3, 4, 10], 
[1, 5, 10], [2, 5, 10], [3, 5, 10], 
[1, 6, 10], [2, 6, 10], [3, 6, 10]]
rnso
źródło
-1
from itertools import product 
list_vals = [['Brand Acronym:CBIQ', 'Brand Acronym :KMEFIC'],['Brand Country:DXB','Brand Country:BH']]
list(product(*list_vals))

Wynik:

[(„Akronim marki: CBIQ”, „Kraj marki: DXB”),
(„Akronim marki: CBIQ”, „Kraj marki: BH”),
(„Akronim marki: KMEFIC”, „Kraj marki: DXB”),
( „Skrót marki: KMEFIC”, „Kraj marki: BH”)]

Kez
źródło
Ta odpowiedź powinna zostać zaakceptowana, ponieważ jako jedyna korzysta z wbudowanej funkcji, podkreślając jednocześnie, że działa również dla każdego, a także heterogenicznego typu.
pedjjj
Czym różni się ta odpowiedź od tej udzielonej wiele lat temu?
Dawid Laszuk