使用 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 中的组合功能来使用 list a 一次，b 2 次不替换，c 2 次不替换，d 和 e 各一次？

[[[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

您想要实现的是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]]

Answer 3

更新了预期输出的说明：

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）。