我的python列表展平代码有什么问题？

Question

我不知道我的代码有什么问题。我不断收到无限循环错误。 我的代码是：

def flatten(aList):   
    x = 0 
    while any_list_in(aList): 
     fList = [] 
     for i in aList: 
      fList.append(i) 
      aList = fList 
    else: 
     return aList 

aList = [[1,'a',['cat'],2],[[[3]],'dog'],4,5] 
def any_list_in(obj): 
    return any(isinstance(item, list) for item in obj) 
flatten(aList) 

Answer 1

这可能是你想要的。 “flatten”创建一个“新列表”并使用帮助函数递归地将原始列表平铺到新列表中。

def flatten_impl(item, newlist): 
    if type(item) == list: 
     for i in item: 
      flatten_impl(i, newlist) 
    else: 
     newlist.append(item) 

def flatten(aList): 
    newlist = [] 
    flatten_impl(aList, newlist) 
    return newlist 

Answer 2

这里有一个flatten_list功能，保留排序，并支持任意嵌套列表，以及非列表项：

def flatten_list(obj): 
    flat_list = [] 
    for item in obj: 
     if isinstance(item, list): 
      flat_list += flatten_list(item) 
     else: 
      flat_list.append(item) 
    return flat_list 

输出：

>>> aList = [[1, 'a', ['cat'], 2], [[[3]], 'dog'], 4, 5, [[[6,7],8],9]] 
>>> flatten_list(alist) 
[1, 'a', 'cat', 2, 3, 'dog', 4, 5, 6, 7, 8, 9] 

Answer 3

我不打算尝试诊断代码中的错误，但这里是一个备用解决方案。

我会使用yield建立一个返回值，而不需要将它们添加到任何类型的列表中。如果您在进行检查时不需要功能any_list_in。如果它是一个列表，那么你使用递归来处理该列表，否则只是从当前迭代中产生每个值。

def flatten(iterable): 
    for i in iterable: 
     if isinstance(i, list): 
      # python 3.4+ can take out the loop 
      #yield from flatten(i) 
      for f in flatten(i): 
       yield f 
     else: 
      yield i 

aList = [[1,'a',['cat'],2],[[[3]],'dog'],4,5] 

print(list(flatten(aList))) 

如果您不能使用yield（可怜的你），那么你只需要建立值的列表，并返回（左作为练习阅读器）。