前一段去云英面试,技术官很”不厚道“了问了一个非常简单的问题:如何将多维列表转化了一维的?当时虽然想到了使用迭代或者列表生成式可以做到,但是可以没能可行的代码,回来后一顿后悔。
对于规范的且嵌套维度较低的多维列表,python中有很多方法可以实现:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
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)
|
对于复杂的多维嵌套列表就需要使用一些复杂的方法:
1
2
3
4
5
6
7
8
9
10
11
12
|
#递归的方法比较容易理解
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
#在*看到大牛的列表生成式版本
func = lambda x: [y for l in x for y in func(l)] if type (x) is list else [x]
#生成式的方法还有很多,可以自行摸索
|
对于使用python2.X的童鞋,还可以使用flatten函数来做:
1
2
3
4
|
>>> 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嵌套列表转一维的方法(压平嵌套列表)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/ACMer_Ding/article/details/73195260