如何在 Python 中扩展列表理解？

如何解决如何在 Python 中扩展列表理解？

我知道在像 [i for j in k for i in j] 这样的列表理解的正常格式中，这基本上可以扩展为：

empty_list = []    
for j in k:
    for i in j:
        empty_list.append(i)

但是，我在递归函数中遇到了一个问题，例如使用递归展平嵌套列表：

def flatten(l:List):
    if isinstance(l,list):
        return [i for j in l for i in flatten(j)]
    else: return [l]

如果我只是扩展这个列表推导式，flatten(j) 将在 nonetype 中：

for j in l:
    for i in flatten(j):
        empty_list.append(i)

这将用完 'nonetype' object is not iterable.

如何正确展开[i for j in l for i in flatten(j)]？谢谢。

解决方法

这是因为在您的第二个示例中，您没有返回列表，而是附加到了一个全局变量。为此，您需要以下内容：

def flatten(l):
    if isinstance(l,list):
        empty_list = []
        for j in l:
            for i in flatten(j):
                empty_list.append(i)
        return empty_list
    else:
        return [l]

实际上，我不会使用任何一种方法。您正在创建一堆昂贵的丢弃容器。相反，我会使用生成器方法：

def flatten(l):
    if isinstance(l,list):
        for j in l:
            yield from flatten(j)
    else:
        yield l

data = [1,[2,[3]]]
print(list(flatten(data)))

[1,2,3]