Python问题——递归组合函数的lambda版本

Question

好的，所以我有以下简单的 Python 3 函数，它完全可以做它应该做的事情：

def comb(a):
    if len(a)==0:
        return[[]]
    else:
        p=[]
        for c in comb(a[1:]):
            p+=[c,c+a[:1]]      
        return p

正确产生

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

出于更好地理解递归函数的愿望，我试图看看是否可以将这个简单的函数转换为 lambda。我尝试的第一件事是各种版本：

comb=lambda a:[[]]if len(a)==0 else[[c,c+a[:1]]for c in comb(a[1:])]

显然，这不起作用。然后我退后一步，只是尝试更改原始函数的语法，例如：

def comb(a):
    return[[]]if len(a)==0 else[p for p in[[c,c+a[:1]]for c in comb(a[1:])]]

这也不起作用，就像根本一样。我尝试了各种不同的安排，但无济于事。显然，原始函数是最有用且最容易阅读的，我只是询问这个以更好地理解我自己的 python。当不涉及递归时，我通常很擅长这个游戏，所以我只是想了解我在这里缺少什么。

Answer 1

如果您的目标是将函数重写为 lambda one-liner，您可以使用 sum() 来实现，例如：

comb = lambda a: [[]] if len(a)==0 else sum(([c,c+a[:1]] for c in comb(a[1:])), [])

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

打印：

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

注意：上述解决方案的可读性不是很高。经典功能更好。

---

感谢@Chris_Rands，itertools.chain 的解决方案：

from itertools import chain

comb = lambda a: [[]] if len(a)==0 else [*chain.from_iterable([c,c+a[:1]] for c in comb(a[1:]))]

Answer 2

您的问题在于列表理解。

[[c,c+a[:1]]for c in l] 是一个长度为 len(l) 的列表，其元素是包含 2 个元素的列表。你想要的是一个长度为2*len(l) 的列表，这是它的扁平化版本。

comb=lambda a:[[]]if len(a)==0 else[c+a[:1]for c in comb(a[1:])] + comb(a[1:])

有效，但可能需要改进，因为每次只需要 1 次时，它都会调用 2 次梳子。