这个function的目的是要unnest list
比如现在我输入:unnest([[[3, 0], 1], 4, 2])。得到的结果是[1,4,2]。正确答案应该是[3,0,1,4,2]
14.15行是我想用recursion做,但是这块好像有问题,求指导。
使用递归的话后面就不用再判断了,递归回来的列表要加在现有的列表里:
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]