python 展开嵌套列表

python 展开嵌套列表

引言

Python中的列表还可以将不同数据类型的项放在一个列表中。所以，一个嵌套的列表就是是一个包含多个列表的列表，例如[1,2，[3]，[4，[5,6]]。

通常，我们需要将这些嵌套列表转换为平面列表（flatted a nested list），以便对数据执行常规列表操作。

方法1,创建递归函数

首先介绍，python中对一个嵌套的list，和一个空列表[]，进行sum()函数运算时，可以实现展开列表。

>>> list_1 = [[1,2],[3]]
>>> sum(list_1, [])
[1, 2, 3]

于是，结合递归原理，可以创建展平嵌套列表的函数如下：

def flatten(li):
    return sum(([x] if not isinstance(x, list) else flatten(x) for x in li), [])

>>> list_1 = [1, 2, [3], [4, [5, 6]]]
>>> print(flatten(list_1))
[1, 2, 3, 4, 5, 6]

方法2,使用列表推导式

>>> li = [[14], [215, 383, 87], [298], [374], [2, 3, 4, 5, 6, 7]]
>>> flat_list = [item for sublist in li for item in sublist]
>>> print(flat_list)
[14, 215, 383, 87, 298, 374, 2, 3, 4, 5, 6, 7]

注意，对于嵌套的列表使用列表推导式时，两个遍历变量的写法要写正确，可以先按照逻辑写推导式中遍历的部分，比如，对于上面的推导式，首先按照逻辑，写遍历的部分是， for sublist in li ；然后继续按照逻辑往下写下一层遍历， for item in sublist。最后，按照逻辑知道，遍历到sublist中的item 就是最后要推导出来的新列表的元素，所以就在列表推导式的最前面加上item。最终组成了完整的嵌套列表的列表推导式。

还要注意的是，这种使用列表推导式展平列表的方法，要求必须是严格嵌套列表，也就是说列表中的每个元素都是列表，而没有非列表的元素，比如下例就会报错:

>>> li = [14, [215, 383, 87], [298], [374], [2, 3, 4, 5, 6, 7]]
>>> flat_list = [item for sublist in li for item in sublist]
Traceback (most recent call last):
  File "<pyshell#77>", line 1, in <module>
    flat_list = [item for sublist in li for item in sublist]
  File "<pyshell#77>", line 1, in <listcomp>
    flat_list = [item for sublist in li for item in sublist]
TypeError: 'int' object is not iterable

这个时候，第一个方法是可以正常工作的。

>>> flatten(li)
[14, 215, 383, 87, 298, 374, 2, 3, 4, 5, 6, 7]

其他的方法

其他的方法有，Flatten List using Inbuilt reduce Function ，Flatten List Using NumPy，请参考Convert Nested List To A Flat List In Python