python列表嵌套合并成一个列表

场景描述

当我们在数据处理，使用列表推导式结合正则匹配时，会产生一个列表中嵌套多个列表的情况。这样遍历列表元素的时候会比较麻烦。因此将所有元素放到一个列表中。

代码处理

1. 方法一：

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

2. 方法二：

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]