我知道可以使用 itertools.permutations
来获取指定长度的所有排列。
但是,对于 itertools.permutations([1,2,3,4],3)
它会返回 (1,2,3)
和 (1,3,2)
。
我想过滤掉这些重复项(即获得组合)。
是否有一种简单的方法来获得所有排列(包括各种长度)?
如何将
itertools.permutation()
的结果转换为常规列表?
我知道可以使用 itertools.permutations
来获取指定长度的所有排列。
但是,对于 itertools.permutations([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)]
看起来您实际上在寻找 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)]
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)]