python 如何处理列表里的小列表

---

问题描述

实际遇到的问题是这样的：有一列表，列表包含几个小列表，形式如下

[[1,2,3,4], [5,6,7,],[8,9,10]]

如何处理使其成为：[1,2,3,4,5,6,7,8,9,10]

---

方法一

利用python的标准库itertools里的函数，chain.from_iterable()

from itertools import chain

l1 = [[1,2,3,4], [5,6,7,],[8,9,10]]

l2 = list(chain.from_iterable(l1))

print(l2)

注意：由于chain.from_iterable()函数结果返回的是迭代器对象，因此需要再list一下。为了解决列表的问题发现itertools这个标准库真心很实用，推荐哦！

方法二

利用降维函数flatten()

flatten是numpy.ndarray.flatten的一个函数，即返回一个一维数组。

flatten只能适用于numpy对象，即array或者mat，普通的list列表不适用。

a.flatten()：a是个数组，a.flatten()就是把a降到一维，默认是按行的方向降 。

import numpy as np

arr1 = [[1,2],[1,2]]

arr2 = list(np.array(arr1).flatten())

print(arr2)

注意：由于flatten()是一个降维函数，因此列表里的小列表的维度必须要一样，像方法一里的例子，flatten()就无法处理。