当存在递归时保留局部变量的当前值

Question

我正在编写一个函数来展平嵌套数组（Pythonlist）。例如将[1,2，[3]]变成[1,2,3]，[[1,2，[3]]，4]变成[1,2,3,4]等。

我有以下内容：

def flatten_array(array):
    flattened_array = []
    for item in array:
        if not isinstance(item, list):
            flattened_array.append(item)
        else:
            flatten_array(item)
    return flattened_array

所以这个想法是让函数是递归的，以处理嵌套到未知深度的情况。我的问题是flattened_array每次遇到嵌套列表都会被重新初始化（当flatten_array被递归调用时）。

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

在进行递归调用时，如何保持展平数组的状态？

Answer 1

更改线条

else:
    flatten_array(item)

到

else:
    flattened_array+=flatten_array(item)

所以完整函数的内容如下

def flatten_array(array):
    flattened_array = []
    for item in array:
        if not isinstance(item, list):
            flattened_array.append(item)
        else:
            flattened_array+=flatten_array(item)
    return flattened_array

这给

flatten_array([1,2,[3]]) # [1,2,3]
flatten_array([1,2,[3,[4,5]]]) # [1,2,3,4,5]
flatten_array([1,2,[3,[4,5]],6,7,[8]]) # [1,2,3,4,5,6,7,8]

您的原始代码对递归调用没有任何作用。您可以在列表中取回结果，但只需丢弃它。我们要做的是将其附加到现有列表的末尾。

此外，如果您不想继续创建临时数组，我们可以在第一次调用函数时创建一个数组，然后附加到它。

def flatten_array(array,flattened_array=None):
    if flattened_array is None:
        flattened_array = []
    for item in array:
        if not isinstance(item,list):
            flattened_array.append(item)
        else:
            flatten_array(item,flattened_array)
    return flattened_array

此版本的结果是相同的，可以以相同的方式使用，但在原始版本中，对函数的每次调用都会创建一个新的空数组来使用。通常这不是问题，但这取决于深度或子数组在内存中的大小。

此版本将阵列展平为给定阵列。仅使用输入调用时（如展平数组（[1,2，[3]）），它会创建一个空数组来处理，否则它只会添加到给定的数组中（因此递归调用只需要提供要添加的数组），并在适当的位置对其进行修改。

如果我们需要，这具有允许您添加到现有数组的优势：

a = [1,2,3]
b = [2,3,[4]] # we want to add flatten this to the end of a
flatten_array(b,a) # we don't bother catching the return result here
print(a) # [1,2,3,2,3,4]

这可能不是我们想要的。通过将默认值设置为“无”，并每次在函数中创建一个新的空数组，我们可以确保对函数的每次调用都有一个唯一的空数组。