如何获取列表的所有组合？答案

【问题标题】：How to get all combinations of a list?如何获取列表的所有组合？
【发布时间】：2013-06-15 03:22:12
【问题描述】：

我知道我可以使用itertools.permutation 来获得大小为 r 的所有排列。

但是，对于itertools.permutation([1,2,3,4],3)，它将返回(1,2,3) 以及(1,3,2)。

我想过滤那些重复（即获得组合）
有没有一种简单的方法来获得所有排列（所有长度）？
如何将itertools.permutation() 结果转换为常规列表？

【问题讨论】：

您在寻找combinations吗？

标签： python combinations permutation itertools

【解决方案1】：

使用itertools.combinations 和一个简单的循环来获得所有大小的组合。

combinations 返回一个迭代器，因此您必须将其传递给 list() 以查看它的内容（或使用它）。

>>> from itertools import combinations
>>> lis = [1, 2, 3, 4]
for i in xrange(1, len(lis) + 1):  #  xrange will return the values 1,2,3,4 in this loop
    print list(combinations(lis, i))
...     
[(1,), (2,), (3,), (4,)]
[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]
[(1,2,3,4)]

【讨论】：

【解决方案2】：

听起来你实际上是在寻找itertools.combinations()：

>>> from itertools import combinations
>>> list(combinations([1, 2, 3, 4], 3))
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]

此示例还展示了如何将结果转换为常规列表，只需将其传递给内置的list() 函数即可。

要获得每个长度的组合，您可以使用如下循环：

>>> data = [1, 2, 3, 4]
>>> for i in range(1, len(data)+1):
...     print list(combinations(data, i))
... 
[(1,), (2,), (3,), (4,)]
[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]
[(1, 2, 3, 4)]

或者要获得嵌套列表的结果，您可以使用列表推导：

>>> [list(combinations(data, i)) for i in range(1, len(data)+1)]
[[(1,), (2,), (3,), (4,)], [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)], [(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)], [(1, 2, 3, 4)]]

对于平面列表而不是嵌套列表：

>>> [c for i in range(1, len(data)+1) for c in combinations(data, i)]
[(1,), (2,), (3,), (4,), (1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4), (1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4), (1, 2, 3, 4)]

【讨论】：

【解决方案3】：

您需要itertools.combinations()。要获取常规列表，只需使用list() 工厂函数即可。

>>> from itertools import combinations
>>> list(combinations([1, 2, 3, 4], 3))
[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]

【讨论】：