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安装目录,设置环境路径以及在命令行运行python脚本实例
这篇文章主要介绍了找Python安装目录,设置环境路径以及在命令行运行python脚本实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-03-03pandas DataFrame 警告(SettingWithCopyWarning)的解决
这篇文章主要介绍了pandas DataFrame 警告(SettingWithCopyWarning)的解决,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2019-07-07tensorflow -gpu安装方法(不用自己装cuda,cdnn)
这篇文章主要介绍了tensorflow -gpu安装,史上最新最简单的途径(不用自己装cuda,cdnn),非常不错,具有一定的参考借鉴价值,需要的朋友可以参考下2020-01-01python opencv鼠标画点之cv2.drawMarker()函数
这篇文章主要给大家介绍了关于python opencv鼠标画点之cv2.drawMarker()函数的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用opencv具有一定的参考下学习价值,需要的朋友可以参考下2021-10-10
最新评论