生成列表的所有可能组合,";itertools。组合&Quot;会丢失一些结果
问题描述
给定Python中的项目列表,我如何才能获得这些项目的所有可能组合?
此站点上有几个类似的问题,建议使用itertools.combinations
,但这只返回我所需的子集:
stuff = [1, 2, 3]
for L in range(0, len(stuff)+1):
for subset in itertools.combinations(stuff, L):
print(subset)
()
(1,)
(2,)
(3,)
(1, 2)
(1, 3)
(2, 3)
(1, 2, 3)
如您所见,它只返回严格顺序的项,而不返回(2, 1)
、(3, 2)
、(3, 1)
、(2, 1, 3)
、(3, 1, 2)
、(2, 3, 1)
和(3, 2, 1)
。有什么办法可以解决这个问题吗?我似乎什么也想不出来。
解决方案
使用itertools.permutations
:
>>> import itertools
>>> stuff = [1, 2, 3]
>>> for L in range(0, len(stuff)+1):
for subset in itertools.permutations(stuff, L):
print(subset)
...
()
(1,)
(2,)
(3,)
(1, 2)
(1, 3)
(2, 1)
(2, 3)
(3, 1)
....
有关itertools.permutations
的帮助:
permutations(iterable[, r]) --> permutations object
Return successive r-length permutations of elements in the iterable.
permutations(range(3), 2) --> (0,1), (0,2), (1,0), (1,2), (2,0), (2,1)
相关文章