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

Question

我知道可以使用 itertools.permutations 来获取指定长度的所有排列。

但是，对于 itertools.permutations([1,2,3,4],3) 它会返回 (1,2,3) 和 (1,3,2)。

1. 我想过滤掉这些重复项（即获得组合）。

2. 是否有一种简单的方法来获得所有排列（包括各种长度）？

3. 如何将 itertools.permutation() 的结果转换为常规列表？

- Bush

---

你在寻找“组合”吗？ - Sukrit Kalra

Answer 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)]

- Ashwini Chaudhary

Answer 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)]

- Andrew Clark

Answer 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)]

- Rohit Jain