代码之家 › 专栏 › 技术社区 › user469652

Python:优化这个循环

for-loop python

user469652 · 技术社区 · 14 年前

a = [(1,2),(3,1),(4,4),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5),(5,5)]
# Quite a lot tuples in the list, 6 digits~
# I want to split it into rows and columns.
rows = 5
cols = 5

Data structure is
rows and cols are the index for the bit list
[rows, cols, (data)]

我使用循环来完成这项工作,但是处理大量元组需要很长时间。

processed_data = []
index = 0
for h in range(0, rows - 1):
    for w in range(0, cols - 1):
        li = []
        li = [h, w, a[index]]
        processed_data.append(li)
        index += 1

这个操作太长了,有没有优化的方法?非常感谢!

3 回复 | 直到 13 年前

aaronasterling 14 年前

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

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。

Ignacio Vazquez-Abrams 14 年前

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

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

Community CDub 8 年前

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

推荐文章

Dave · 如何在for循环中修改列表值

7 月前

Haru Hoshizora · 为什么一个整数的位置没有改变,但值却不同

8 月前

BlurKid · R中for循环时结果的奇怪差异

8 月前

Rudraksh_pd · 取炭。通过char。c中创建字符串的输入

8 月前

Mtullis · 在我的表单值中循环遍历数组[重复]

9 月前

puboot · 我的for循环没有运行,我不知道为什么。它甚至不会在控制台上打印任何内容

1 年前

Justin Hawkins · 在一个数组中返回两个数组,其中包含带特定字母的名称和不带指定字母的名称

1 年前

leiseg · 使用扩展变量的for循环中PowerShell CLI中缺少终止符

1 年前

xhamsterIT · 循环VBA Microsoft Excel

1 年前

AlexC · 如何循环dplyr group_by并总结变量列表的语句

1 年前