python中很多时候可能会遇到多层嵌套可迭代列表如下:
list1 = [11, 22, [33, 44, [55, 66]]]
list2 = [11, 22, [33, 44]]
其中我们想要的结果只是一个所有的结果列表而已.此时就需要一个剥皮函数,把嵌套的元素全部剥离出来,放入一个列表中(递归实现).
实现函数如下:
def f(x, ret=None):
# 把上一次的结果传递过来
if not ret:
ret = []
for i in x:
if isinstance(i, list):
# 如果i为列表,则返回继续处理,且把现在的处理结果传递回去
f(i, ret)
else:
ret.append(i)
# 最终返回所有的元素
return ret
第二种方法:
# 第二种方法
def f2(x):
ret = []
for m in x:
if isinstance(m, list):
# 进行递归剥皮将列表挨个递归添加到ret列表中
# 因为f2函数最终返回的是一个列表,所以不管多少层f2(m)将始终是一个列表
# 循环将元素加入ret列表中
for n in f2(m):
ret.append(n)
else:
ret.append(m)
return ret
原文地址:https://blog.51cto.com/smly1989/2402275
时间: 2024-10-29 04:25:09