生成列表的所有可能组合,"itertools。组合&Quot;会丢失一些结果

2022-03-23 00:00:00 python list combinations

问题描述

给定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)

相关文章