Создание комбинированных из списков с помощью itertools

Question

import itertools 
a = [[2, 3], [3, 4]] 
b = [[5, 6], [7, 8], [9, 10]] 
c = [[11, 12], [13, 14]] 
d = [[15, 16], [17, 18]] 
e = [[12,16],[13,17],[14,18],[15,19]] 

q=[] 
q=list(itertools.combinations((a, b, b,c, c, d,e),7) 
print q 

Как бы я идти об использовании функции комбинации из itertools правильно использовать список один раз, а в 2 раза без замены, C 2 раза без замены, и d и е один раз каждый ?

[[[2, 3],[5, 6],[7, 8],[11, 12],[13, 14],[15, 16],[12,16]], 
[[2, 3],[5, 6],[7, 8],[11, 12],[13, 14],[15, 16],[13,17]], 
[[2, 3],[5, 6],[7, 8],[11, 12],[13, 14],[15, 16],[14,18]], 
[[2, 3],[5, 6],[7, 8],[11, 12],[13, 14],[15, 16],[15,19]], 
[[2, 3],[5, 6],[7, 8],[11, 12],[13, 14],[15, 16],[12,16]],... 
[[3, 4],[7, 8],[9, 10],[11, 12], [13, 14],[17, 18],[15,19]]] 

Answer 1

Похоже, вы ищете сочетание combinations и product: Используйте combinations, чтобы получить возможные комбинации без замены повторяющихся списков, а затем использовать product, чтобы объединить все эти комбинации. Вы можете поместить списки и подсчеты в два списка, zip этих списков и использовать выражение генератора для получения всех комбинаций.

from itertools import product, combinations, chain 
lists = [a,b,c,d,e] 
counts = [1,2,2,1,1] 
combs = product(*(combinations(l, c) for l, c in zip(lists, counts))) 

Для этого, например, combs генератора имеет 48 элементов, среди прочего:

[(([2, 3],), ([5, 6], [7, 8]), ([11, 12], [13, 14]), ([15, 16],), ([12, 16],)), 
... 
(([2, 3],), ([5, 6], [7, 8]), ([11, 12], [13, 14]), ([17, 18],), ([15, 19],)), 
(([2, 3],), ([5, 6], [9, 10]),([11, 12], [13, 14]), ([15, 16],), ([12, 16],)), 
... 
(([3, 4],), ([5, 6], [7, 8]), ([11, 12], [13, 14]), ([15, 16],), ([12, 16],)), 
... 
(([3, 4],), ([5, 6], [7, 8]), ([11, 12], [13, 14]), ([17, 18],), ([15, 19],)), 
... 
(([3, 4],), ([7, 8], [9, 10]),([11, 12], [13, 14]), ([17, 18],), ([15, 19],))] 

Если вы хотите flattened lists, просто chain им:

>>> combs = (list(chain(*p)) for p in product(*(combinations(l, c) for l, c in zip(lists, counts)))) 
>>> list(combs) 
[[[2, 3], [5, 6], [7, 8], [11, 12], [13, 14], [15, 16], [12, 16]], 
... 
[[3, 4], [7, 8], [9, 10], [11, 12], [13, 14], [17, 18], [15, 19]]] 

Answer 2

Обновлено дано разъяснение ожидаемого выхода:

You want itertools.product:

itertools.product(a, b, b, c, c, c, c, d, e) 

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

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

itertools.product(a, *[b]*2, *[c]*4, d, e) 

Или использовать tobias_k's solution для более общего повторения последовательностей (которые также будут работать на py2).

Answer 3

Что ваше пытаются достичь это Cartesian product of input iterables, а не комбинации элемента, присутствующего в списке. Следовательно, вместо этого вы должны использовать itertools.product().

В случае повторения допускается из списков, которые используются более чем один раз, ответ прост:

>>> import itertools 
>>> a = [1,2] 
>>> b = [3,4] 
>>> [i for i in itertools.product(a, b, b)] 
[(1, 3, 3), (1, 3, 4), (1, 4, 3), (1, 4, 4), (2, 3, 3), (2, 3, 4), (2, 4, 3), (2, 4, 4)] 

Но в случае repetition is not allowed в тех же списках, он будет немного противно, и вы должны объединить выше ответьте combinations() и chain() (то же, что указано tobias_k). Этот код даст список всех combinations:

>>> from itertools import chain, product, combinations 
>>> lists, counts = [a, b], [1, 2] # to track that a is to be used once, and b twice 
>>> list(list(chain(*p)) for p in product(*(combinations(l, c) for l, c in zip(lists, counts)))) 
[[1, 3, 4], [2, 3, 4]] 

Однако, в случае, если вам нужны перестановки вместо комбинаций, вы должны обновить код выше с permutations():

>>> from itertools import chain, product, permutations 
>>> list(list(chain(*p)) for p in product(*(permutations(l, c) for l, c in zip(lists, counts)))) 
[[1, 3, 4], [1, 4, 3], [2, 3, 4], [2, 4, 3]]