多维列表(元组)碾成一维形式
一.需求
原格式:
input=[[1,2,3],[4,5,6],[7,8,9]]
目标格式:
[1, 2, 3, 4, 5, 6, 7, 8, 9]
二.方法
1.sum函数合并
input=[[1,2,3],[4,5,6],[7,8,9]] output=sum(input,[]) print(output) #结果: [1, 2, 3, 4, 5, 6, 7, 8, 9]
这个看上去很简洁,不过有类似字符串累加的性能陷阱。
2.reduce函数
from functools import reduce input=[[1,2,3],[4,5,6],[7,8,9]] output=reduce(list.__add__, input) print(output) #结果
[1, 2, 3, 4, 5, 6, 7, 8, 9]
做序列的累加操作。也是有累加的性能陷阱。
3.列表推导式
input=[[1,2,3],[4,5,6],[7,8,9]] output=[item for sublist in input for item in sublist] print(output) #结果 [1, 2, 3, 4, 5, 6, 7, 8, 9]
列表推导式,看着有些长,而且还要for循环两次,变成一行理解需要费劲一些,没有那么直观
4.itertools 类库
import itertools input=[[1,2,3],[4,5,6],[7,8,9]] ouput=list(itertools.chain(*input)) print(ouput) #结果 [1, 2, 3, 4, 5, 6, 7, 8, 9]
三.性能对比
python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' '[item for sublist in l for item in sublist]' 10000 loops, best of 3: 51.2 usec per loop python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'reduce(list.__add__,l)' 1000 loops, best of 3: 572 usec per loop python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'sum(l, [])' 1000 loops, best of 3: 545 usec per loop python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99; import itertools;' 'list(itertools.chain(*l))' 10000 loops, best of 3: 35.1 usec per loop