遍历一个嵌套列表

Question

我正在尝试使用递归遍历列表并将其所有非列表元素和子列表的元素附加到唯一列表。

def refr(a_list):
    loo = []
    for item in a_list:
        if isinstance(item, list):
            refr(item)
        else:
            loo.append(item)
    return loo

print(refr([1, ['no', 'ok'], 'dang', 0]))

当我将 loo 定义为局部变量时，它只包含输入列表中不代表列表的元素。

# output

[1, 'dang', 0]

但是当定义为全局时，输出是正确的

# output

[1, 'no', 'ok', 'dang', 0]

为什么会这样？

Answer 1

使用来自document的extend(item)：

list.extend（可迭代） 通过附加迭代中的所有项目来扩展列表。等价于 a[len(a):] = 可迭代。

试试这个：

def refr(a_list):
    loo = []
    for item in a_list:
        if isinstance(item, list):
            loo.extend(refr(item))
            # OR
            # loo += refr(item)
        else:
            loo.append(item)
    return loo

print(refr([1, ['no', 'ok'], 'dang', 0]))

输出：

[1, 'no', 'ok', 'dang', 0]

对于这个输入得到正确的输出：

print(refr([1, [['no', 'ok'], ['dang', 0]]]))
# [1, 'no', 'ok', 'dang', 0]

Answer 2

您没有使用递归调用的返回值：

def refr(a_list):
    loo = []
    for item in a_list:
        if isinstance(item, list):
            loo += refr(item)
        else:
            loo.append(item)
    return loo

或者，使用生成器函数，这通常就足够了，允许短路并且可以很容易地变成一个列表：

def refr(a_list):
    for item in a_list:
        if isinstance(item, list):
            yield from refr(item)
        else:
            yield item

从这里到单线不远，使用itertools.chain：

from itertools import chain

def refr(a_list):
    return chain.from_iterable(refr(x) if isinstance(x, list) else [x] for x in a_list)

[*refr([1, ['no', 'ok', [1, 2, [3]]], 'dang', 0])]
# [1, 'no', 'ok', 1, 2, 3, 'dang', 0]

Answer 3

因为本地 (!) loo 变量的范围是 refr 函数用于该特定调用。这意味着当您递归调用 refr 时，它将创建一个新的堆栈框架，其中包含另一个与第一个无关的空列表（它恰好具有相同的名称）。解决此问题的一种方法是在您的函数调用中添加一个累加器参数作为组合状态：

def refr(a_list, acc=None):
    if acc is None:
        acc = []
    for item in a_list:
        if isinstance(item, list):
            refr(item, acc)
        else:
            acc.append(item)
    return acc

refr([1, ['no', 'ok'], 'dang', 0])

给予

[1, 'no', 'ok', 'dang', 0]