python-将组合函数转换为生成器

Question

我有一个函数可以创建列表项的所有组合，并将其表示为列表列表：

def makeCombos(arr):
  yield (sum([map(list, combinations(arr, i)) for i in range(len(arr) + 1)], []))

调用makeCombos([1,2,3,4,5])给了我一个生成器对象，但是调用.next()没有给我一个连击，而是给了我整个连击列表。

如何将其转换为可以调用的生成器函数？

Answer 1

sum(iterable, [])不会创建列表列表。 它实际上使事情变得平坦。

在此行中的yield (sum(...))仅产生一个项目，即所有组合的扁平化列表。

对于Python 2.X， sum([(map(list, combinations(arr, i))) ...])可以使用，但是在Python 3.X中map不再返回列表。 而是返回一个地图对象。 因此，如果有人使用Python 3.X，只需将其转换为list(map(.....)) ，即可在3.X上运行。

我认为您真正想要的是这样的：

from itertools import combinations
def makeCombos(arr):
     for i in range(len(arr) + 1):
         for combo in map(list, combinations(arr, i)):
            yield combo

#Or call next 

combos = makeCombos([1, 2, 3, 4, 5])
for combo in combos:
     print combo

注释的替代方案：

除了让步，我们可以像返回 yield一样返回生成器对象并循环遍历。

例如-

from itertools import combinations

def makeCombos(arr):

     return (combo for i in range(len(arr) + 1) for combo in map(list, combinations(arr, i)))

combos = makeCombos([1, 2, 3, 4, 5])
....

至于这是“ Pythonic”，我不会这么说。 我实际上更喜欢嵌套的forloop，它更具可读性。

虽然，我们仍然可以尝试做一些“技巧”来清理/压缩它

from itertools import combinations as cs #or some other name)

def makeCombos(arr):

    return (c for i in range(len(arr) + 1) for c in map(list, cs(arr, i)))

但是，现在您已经失去了所有可读性，这就像您在Perl中看到的一样。 （惊恐的事件！）

输出：

[]
[1]
[2]
[3]
[4]
[5]
[1, 2]
[1, 3]
[1, 4]
[1, 5]
[2, 3]
[2, 4]
[2, 5]
[3, 4]
[3, 5]
[4, 5]
[1, 2, 3]
[1, 2, 4]
[1, 2, 5]
[1, 3, 4]
[1, 3, 5]
[1, 4, 5]
[2, 3, 4]
[2, 3, 5]
[2, 4, 5]
[3, 4, 5]
[1, 2, 3, 4]
[1, 2, 3, 5]
[1, 2, 4, 5]
[1, 3, 4, 5]
[2, 3, 4, 5]
[1, 2, 3, 4, 5]

Answer 2

itertools已经有一种将可迭代对象连接在一起的方法：称为chain 。 您想要的是以下内容：

def makeCombos(arr):
    return chain.from_iterable(combinations(arr, i) for i in range(len(arr) + 1))

在我看来，它简单，简短且相当Pythonic。