Python嵌套列表转一维的方法(压平嵌套列表)
前一段去云英面试,技术官很”不厚道“了问了一个非常简单的问题:如何将多维列表转化了一维的?当时虽然想到了使用迭代或者列表生成式可以做到,但是可以没能可行的代码,回来后一顿后悔。
对于规范的且嵌套维度较低的多维列表,python中有很多方法可以实现:
a=[[1,2],[3,4],[5,6]] print[jforiinliforjini] #or fromitertoolsimportchain printlist(chain(*a)) #or importitertools 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)foriina] printt #or a=[[1,2],[3,4],[5,6]] printsum(a,[]) #or reduce(lambdax,y:x+y,a)
对于复杂的多维嵌套列表就需要使用一些复杂的方法:
#递归的方法比较容易理解 defexpand_list(nested_list): foriteminnested_list: ifisinstance(item,(list,tuple)): forsub_iteminexpand_list(item): yieldsub_item else: yielditem #在stackoverflow看到大牛的列表生成式版本 func=lambdax:[yforlinxforyinfunc(l)]iftype(x)islistelse[x] #生成式的方法还有很多,可以自行摸索
对于使用python2.X的童鞋,还可以使用flatten函数来做:
>>>d=[1,2,3,[4,5],[6,7,[8,9]]] >>>fromcompiler.astimportflatten >>>flatten(a) [1,2,3,4,5,6,7,8,9]
其实上边列举的方法仅仅是冰山一角,大海下的冰山就等这童鞋们自行去寻找挖掘了。
以上这篇Python嵌套列表转一维的方法(压平嵌套列表)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。