场景描述
当我们在数据处理,使用列表推导式结合正则匹配时,会产生一个列表中嵌套多个列表的情况。这样遍历列表元素的时候会比较麻烦。因此将所有元素放到一个列表中。
代码处理
- 方法一:
def extend_list(arg):
new_list = []
for i in arg:
if not isinstance(i,list):
new_list.append(i)
else:
new_list.extend(extend_list(i))
return new_list
- 方法二:
def sum_list(arg):
return sum(arg, [])
举个例子
>>> my_list = [[1,2,3],[4,5,6],[7,8,9]]
>>> def extend_list(arg):
... new_list = []
... for i in arg:
... if not isinstance(i,list):
... new_list.append(i)
... else:
... new_list.extend(extend_list(i))
... return new_list
...
>>> def sum_list(arg):
... return sum(arg, [])
...
>>> extend_list(my_list)
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> sum_list(my_list)
[1, 2, 3, 4, 5, 6, 7, 8, 9]