题目:
把嵌套的List展平
Example:
flat_list([1, 2, 3]) == [1, 2, 3]
flat_list([1, [2, 2, 2], 4]) == [1, 2, 2, 2, 4]
flat_list([[[2]], [4, [5, 6, [6], 6, 6, 6], 7]]) == [2, 4, 5, 6, 6, 6, 6, 6, 7]
flat_list([-1, [1, [-2], 1], -1]) == [-1, 1, -2, 1, -1]
Precondition: 0 ≤ |array| ≤ 100
∀ x ∈ array : -232 < x < 232 or x is a list
depth < 10
思路1:递归,定义一个函数,如果x里面元素是数字就append,如果不是就递归到下一级
代码1:
def flat_list(x):
r = []
def ff(x):
for i in x:
if type(i) is int:
r.append(i)
else:
ff(i)
ff(x)
return r
思路2:
类似1
代码2:
a = [[[2]], [4, [5, 6, [6], 6, 6, 6], 7]]
f=lambda d:[d] if type(d) == int else sum(map(f,d),[])
print(f(a))
思路3:
正则,把数字选出来
代码3
import re
array = re.findall('\d', str(array))
print(array)
list = []
for i in array:
i = int(i)
list.append(i)
print(list)