一.需求
原格式:
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學習交流群:725638078
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