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

问题描述

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

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

1. 我要过滤这些重复项(即获取组合)

2. 有没有一种简单的方法可以获得(所有长度的)所有排列？

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

---

解决方案

使用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)]