Unnest list中recursion的问题

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

img

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

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]