Question

我需要能够创建一个包含输入列表的所有可能组合的列表。例如，列表[1,2,3]应该返回[1 [1,2] [1,3] 2 [2,3] 3 [1,2,3]] 该列表不必按任何特定顺序排列。在这个网站上我发现了许多使用itertools的函数，但是当我只需要一个list时，它们就会返回对象。

Answer 1

只需使用itertools.combinations即可。例如：

import itertools

lst = [1, 2, 3]
combs = []

for i in xrange(1, len(lst)+1):
    combs.append(i)
    els = [list(x) for x in itertools.combinations(lst, i)]
    combs.append(els)

现在combs拥有此值：

[1, [[1], [2], [3]], 2, [[1, 2], [1, 3], [2, 3]], 3, [[1, 2, 3]]]

是的，它与您提供的示例输出略有不同，但在该输出中您没有列出所有可能的组合。

我列出了每个尺寸的实际列表之前组合的大小，如果您需要的只是组合（没有大小，因为它出现在您的示例输出中）然后尝试这些代码的其他版本：

import itertools lst = [1, 2, 3] combs = [] for i in xrange(1, len(lst)+1): els = [list(x) for x in itertools.combinations(lst, i)] combs.extend(els)

现在combs拥有此值：

[[1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]

Answer 2

itertools模块确实返回生成器而不是列表，但是：

生成器通常比列表更有效（特别是如果要生成大量组合）
您可以随时使用list(...) 将生成器转换为列表。

chain的{{1}}和combinations函数效果不错，但您需要使用 Python 2.6或更高版本： < / p>

itertools

然后你可以这样称呼它：

import itertools

def all_combinations(any_list):
    return itertools.chain.from_iterable(
        itertools.combinations(any_list, i + 1)
        for i in xrange(len(any_list)))

如果您之前没有使用过生成器，请注意您循环遍历它们，就像它们是一个列表一样，如下所示：

# as a generator
all_combinations([1,2,3])  # --> <itertools.chain at 0x10ef7ce10>

# as a list
list(all_combinations([1,2,3]))  # --> [(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]

# as a list of lists
[list(l) for l in all_combinations([1,2,3])]  # --> [[1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]

性能差异可能很大。如果您比较性能，您会发现生成器的创建速度要快得多：

# a generator returned instead of list
my_combinations = all_combinations([1,2,3])

# this would also work if `my_combinations` were a list
for c in my_combinations:
    print "Combo", c

"""
Prints:
  Combo (1,)
  Combo (2,)
  Combo (3,)
  Combo (1, 2)
  Combo (1, 3)
  Combo (2, 3)
  Combo (1, 2, 3)
"""

请注意，在任何一种情况下都需要花费一些时间来迭代所有组合，但对于您来说这可能是一个很大的胜利，特别是如果您在早期找到了您想要的东西。

Answer 3

itertools模块中的函数返回迭代器。要将这些转换为列表，您只需在结果上调用list()。

但是，由于您需要分别调用itertools.combinations三次（每个不同长度一次），您可以使用list.extend将迭代器的所有元素添加到最终列表中。

尝试以下方法：

import itertools
in_list = [1, 2, 3]
out_list = []
for i in range(1, len(in_list)+1):
    out_list.extend(itertools.combinations(in_list, i))

或者作为列表理解：

out_list = [c for i in range(len(in_list)) for c in itertools.combinations(in_list, i+1)]

这些将产生以下列表：

[(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]

如果您想要列表而不是元组，并将单个长度元组转换为该值，则可以执行以下操作：

out_list = [x[0] if len(x) == 1 else list(x) for x in out_list]
# [1, 2, 3, [1, 2], [1, 3], [2, 3], [1, 2, 3]]

或者将单个项目保留为列表：

out_list = map(list, out_list)

Answer 4

您可以在循环内使用itertools.combinations解决问题：

>>> l = [1,2,3]
>>> comb = []
>>> for i in range(len(l)):
...   comb += itertools.combinations(l,i+1)
... 
>>> comb
[(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]

如果你想把它们作为一个清单：

>>> comb_list = [ list(t) for t in comb ]
>>> comb_list
[[1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]

编辑：组合的第一个参数是可迭代的，第二个参数是结果元组的长度（在这种情况下，从1到len(l)）

有关组合的更多信息：http://docs.python.org/library/itertools.html#itertools.combinations

Answer 5

l = [1,2,3]
combs = reduce(lambda x, y: list(itertools.combinations(l, y)) + x, range(len(l)+1), [])

如果你想要一个oneliner。

Answer 6

我认为值得将此处的其他答案归纳为一个简单的Python 3示例：

from itertools import chain, combinations

def all_combinations(array):
    return chain(*(list(combinations(array, i + 1)) for i in range(len(array))))

这将返回一个迭代值，以查看值：

>>> print(list(all_combinations((1, 2, 3))))
[(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]

制作列表的所有可能组合

6 个答案: