Все комбинации списка списков

Question

Я в основном ищет версию питона из Combination of List<List<int>>

Учитывая список списков, мне нужен новый список, который дает все возможные комбинации элементов между списками.

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

Количество списков неизвестно, поэтому мне нужно что-то, что работает для всех случаев. Бонусные очки за элегантность!

Answer 1

вам нужно 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)] 

Answer 2

Самое элегантное решение заключается в использовании itertools.product в Python 2.6.

Если вы не используете Python 2.6, то документы для itertools.product фактически показывают эквивалентную функцию, чтобы сделать продукт «ручной» способ:

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) 

Answer 3

listOLists = [[1,2,3],[4,5,6],[7,8,9,10]] 
for list in itertools.product(*listOLists): 
    print list; 

Я надеюсь, что вы найдете, что, как элегантный, как я, когда я впервые столкнулся с этим.

Answer 4

Numpy может это сделать:

>>> 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], ....] 

Answer 5

Ничего плохого с прямой до рекурсии для выполнения этой задачи, и если вам нужна версия, которая работает со строками, это могло бы соответствовать вашим потребностям:

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