更新:使用扩展但不理解且不使用列表作为迭代器进行展平(最快)
在检查了下一个答案之后,dual for
我通过列表理解提供了一个更快的解决方案,我做了一些调整,现在它表现更好,首先执行list(...)拖了很大的时间,然后更改了列表对一个简单循环的理解也减少了一些。
新的解决方案是:
l = []
for row in output: l.extend(row)
较旧:
使用地图/扩展进行展平:
l = []
list(map(l.extend, output))
使用列表理解而不是地图进行展平
l = []
list(l.extend(row) for row in output)
只需删除[...]的list(...),就可以进行一些新的扩展和改进:
import timeit
t = timeit.timeit
o = "output=list(zip(range(1000000000), range(10000000))); l=[]"
steps_ext = "for row in output: l.extend(row)"
steps_ext_old = "list(l.extend(row) for row in output)"
steps_ext_remove_list = "[l.extend(row) for row in output]"
steps_com = "[item for sublist in output for item in sublist]"
print("new extend: ", t(steps_ext, setup=o, number=10))
print("old extend w []: ", t(steps_ext_remove_list, setup=o, number=10))
print("comprehension: ", t(steps_com, setup=o, number=10,))
print("old extend: ", t(steps_ext_old, setup=o, number=10))
>>> new extend: 4.502427191007882
>>> old extend w []: 5.281140706967562
>>> comprehension: 5.54302118299529
>>> old extend: 6.840151469223201
[(12.2817, 12.2817), (0, 0), (8.52, 8.52)]
已经是3x2矩阵!还是我错过了什么?