Unnest list中recursion的问题

Question

这个function的目的是要unnest list
比如现在我输入：unnest([［[3, 0], 1], 4, 2])。得到的结果是［1，4,2］。正确答案应该是［3,0，1，4,2］
14.15行是我想用recursion做，但是这块好像有问题，求指导。

Answer 1

使用递归的话后面就不用再判断了，递归回来的列表要加在现有的列表里：

def unnest(alist):
    if alist==[]:
        return []
    else:
        answer=[]
        for ele in alist:
            if type(ele)==int:
                answer.append(ele)
            elif type(ele)==list:
                answer += unnest(ele)
    return answer

结果：

print(unnest([[[3,0],1],4,2]))
[3, 0, 1, 4, 2]