python - 在没有 itertools 的情况下使用 python 进行笛卡尔积

Question

我遇到了这行很好的Python代码，用于n个列表的笛卡尔积，我想将其分多行带回来:

def cart_product_1(*seqs):
    if not seqs:
        return [[]]
    else:
        return [[x] + p for x in seqs[0] for p in cart_product_1(*seqs[1:])]

这对我来说似乎很简单，但显然我在这里遗漏了一些东西。我想我需要在某处附加一个列表，但不太明白。

def cart_product_1(result,*seqs):
    if not seqs:
        return [[]]
    else:
        for x in seqs[0]:                       
            for p in cart_product_1(result,*seqs[1:]):
                 result.append([x]+p)
        return result

这会导致内存错误。

Answer 1

示例中的行是 list comprehension 。基本上，一行是构建一个列表并计算其成员。

要做同样的事情，你必须添加一个初始化器并返回:

result = []
for x in seqs[0]:                       
    for p in cart_product_1(*seqs[1:]):
        result.append([x]+p)
return result