Python-获取列表的所有组合
问题内容:
我知道我可以使用itertools.permutation获得大小为r的所有排列。但是,对于itertools.permutation([1,2,3,4],3)
它将返回(1,2,3)
以及(1,3,2)
。
-
我想过滤这些重复(即获得组合)
-
是否有一种简单的方法来获得所有排列(所有长度)?
-
如何将
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)]