Python嵌套列表转一维的方法(压平嵌套列表)
时间：2021-01-13 09:50:35|栏目：Python代码|点击： 次
前一段去云英面试，技术官很”不厚道“了问了一个非常简单的问题：如何将多维列表转化了一维的？当时虽然想到了使用迭代或者列表生成式可以做到，但是可以没能可行的代码，回来后一顿后悔。
对于规范的且嵌套维度较低的多维列表，python中有很多方法可以实现：
a=[[1,2],[3,4],[5,6]] print [j for i in li for j in i] #or from itertools import chain print list(chain(*a)) #or import itertools a = [[1,2,3],[4,5,6], [7], [8,9]] out = list(itertools.chain.from_iterable(a)) #or a=[[1,2],[3,4],[5,6]] t=[] [t.extend(i) for i in a] print t #or a=[[1,2],[3,4],[5,6]] print sum(a,[]) #or reduce(lambda x, y: x+ y, a)
对于复杂的多维嵌套列表就需要使用一些复杂的方法：
#递归的方法比较容易理解
def expand_list(nested_list):
for item in nested_list:
if isinstance(item, (list, tuple)):
for sub_item in expand_list(item):
yield sub_item
else:
yield item
#在stackoverflow看到大牛的列表生成式版本
func = lambda x: [y for l in x for y in func(l)] if type(x) is list else [x]
#生成式的方法还有很多，可以自行摸索
对于使用python2.X的童鞋，还可以使用flatten函数来做：
>>> d=[1,2,3,[4,5],[6,7,[8,9]]] >>>from compiler.ast import flatten >>>flatten(a) [1, 2, 3, 4, 5, 6,7,8,9]
其实上边列举的方法仅仅是冰山一角，大海下的冰山就等这童鞋们自行去寻找挖掘了。
栏 目：Python代码
下一篇：python 调用pyautogui 实时获取鼠标的位置、移动鼠标的方法
本文地址：http://www.codeinn.net/misctech/44445.html
