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

Question

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

Я в основном ищу версию python сочетание List<List<int>>

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

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

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

355 5

python

5 ответов:

Comments

Ничего не найдено.

SilentGhost · Accepted Answer · 2014-09-29 18:22:24

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

Jarret Hardie · Accepted Answer · 2014-09-29 18:23:01

самое элегантное решение-использовать itertools.продукт в python 2.6.

Если вы не используете Python 2.6, документы для itertools.продукт фактически показывает эквивалентную функцию, чтобы сделать продукт "ручным" способом:
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)

Matthew Flaschen · Accepted Answer · 2009-04-28 19:58:13

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

Diamantatos Paraskevas · Accepted Answer · 2017-01-31 04:48:05

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

duanev · Accepted Answer · 2017-01-31 04:56:29

нет ничего плохого в прямой рекурсии для этой задачи, и если вам нужна версия, которая работает со строками, это может соответствовать вашим потребностям:
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']