以递归方式展平Python中的不规则列表列表

Question

我搜索并发现具有相同标题的问题也是（hereherehereherehere）但我不是在问这个问题。我遇到了这个问题：

编写一个函数来展平列表。该列表包含其他列表、字符串或整数。

我的代码是

t=[]
def flatten(aList):
    for i in aList:
        if type(i) !=list:
             t.append(i)
        else:
             flatten(i)

    return t     

但是当我检查测试用例的代码时：

1. flatten([[1], [1]]) ：检查器告诉我输出是[1, 1, 1, 1] 但在 codeskulptor 我得到正确的输出是 [1, 1] 。
2. flatten([[[1]], [[[5]]]])：检查器告诉输出是[1, 1, 1, 1, 1, 2, 3, 3, 2, 1, 0, 4, 5, 6, 7, 1, 5]，但在codeskulptor 中告诉[1, 5]。

很多测试用例都存在这个问题。 然后我在 python 导师中检查了我的代码，发现在每次返回列表 t 时执行 if 语句后，最后当函数停止时，它返回最后编辑的列表 t。

我该如何解决这个问题，请帮我解决这个问题，是的，我是 python 新手，对 itertools、lambda 函数使用、生成器等一无所知。所以请在我能理解的上下文中告诉我。

Answer 1

您的代码依赖于全局；如果检查器两次调用您的函数，它将收到比预期更长的列表：

>>> t = []
>>> def flatten(aList):
...     for i in aList:
...         if type(i) !=list:
...              t.append(i)
...         else:
...              flatten(i)
...     return t
...
>>> flatten([1, 1])
[1, 1]
>>> flatten([1, 1])
[1, 1, 1, 1]
>>> t  # your global, still holding all those values:
[1, 1, 1, 1]

不要使用全局变量。使用本地列表，并使用递归调用的结果对其进行扩展：

def flatten(aList):
    t = []
    for i in aList:
        if not isinstance(i, list):
             t.append(i)
        else:
             t.extend(flatten(i))
    return t

请注意，我切换到使用 isinstance() 来测试类型。此版本不会受到共享状态泄漏到下一次调用中的影响：

>>> def flatten(aList):
...     t = []
...     for i in aList:
...         if not isinstance(i, list):
...              t.append(i)
...         else:
...              t.extend(flatten(i))
...     return t
...
>>> flatten([1, 1])
[1, 1]
>>> flatten([1, 1])
[1, 1]
>>> flatten([[[1]], [[[5]]]])
[1, 5]
>>> flatten([1, 1, [42, 81]])
[1, 1, 42, 81]