Python:优化这个循环

我一点也不清楚你想要什么,但这里有一个以更优化的方式在同一个循环中拍摄的镜头:

import itertools as it

index = it.count(0) 
processed_data = [[h, w, a[next(index)]] 
                 for h in xrange(0, rows - 1)
                 for w in xrange(0, cols - 1)]

index = ite.count(0)
indices = it.product(xrange(0, rows-1), xrange(0, cols-1))
processed_data = [[h, w, a[next(index)]] for h, w in indices]

它们更快的原因是它们使用列表理解而不是 for 循环。列表理解有自己的操作码List_APPEND,它直接路由到 append 对于 循环,虚拟机必须经历查找 追加

另外,itertools是用C语言实现的,所以如果对同一算法来说速度不快,那么itertools中就有一个bug。

好吧,如果你真的非常想要指数。。。

[divmod(i, cols) + (x,) for i, x in itertools.izip(itertools.count(), a)]

听起来你想 split it into evenly-sized chunks .