前一段去云英面試,技術官很”不厚道“了問了一個非常簡單的問題:如何將多維列表轉化了一維的?當時雖然想到了使用迭代或者列表生成式可以做到,但是可以沒能可行的代碼,回來后一頓后悔。
對于規范的且嵌套維度較低的多維列表,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嵌套列表轉一維的方法(壓平嵌套列表)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答