Python嵌套列表转一维的方法(压平嵌套列表)

Python  /  管理员 发布于 7年前   196

前一段去云英面试，技术官很”不厚道“了问了一个非常简单的问题：如何将多维列表转化了一维的？当时虽然想到了使用迭代或者列表生成式可以做到，但是可以没能可行的代码，回来后一顿后悔。

对于规范的且嵌套维度较低的多维列表，python中有很多方法可以实现：

a=[[1,2],[3,4],[5,6]]print [j for i in li for j in i]#orfrom itertools import chainprint list(chain(*a))#orimport itertoolsa = [[1,2,3],[4,5,6], [7], [8,9]]out = list(itertools.chain.from_iterable(a))#ora=[[1,2],[3,4],[5,6]]t=[][t.extend(i) for i in a]print t#ora=[[1,2],[3,4],[5,6]]print sum(a,[])#orreduce(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嵌套列表转一维的方法(压平嵌套列表)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。