python - 如何使字典乘法更快？

Question

有没有办法让函数 f() 更快。我运行了数亿次，所以任何速度的提高都会受到赞赏。字典 a 和 b 的 ID 在运行中是相同的，w 是一个常量。键是整数； key 通常分布不均。

此外，函数在类对象中。所以f就是f(self)，变量是self.w、self.ID、self.a、self.b

w = 10.25
ID = range(10)
a = {}
b = {}

for i in ID:
    a[i] = random.uniform(0,1)
    b[i] = random.uniform(0,1)

def f():
    for i in ID:
        a[i] = b[i] * w

t0 = time.time()
for i in xrange(1000000):
    f()
t1 = time.time()
print t1-t0

Answer 1

您可以通过在 f() 中本地化变量来获得很好的加速:

def f(ID=ID, a=a, b=b, w=w):
    for i in ID:
        a[i] = b[i] * w

有些人不喜欢以这种方式进行本地化，因此您还可以构建一个闭包来加快全局变量访问速度:

def make_f(a, b, w, ID):
    def f():
        for i in ID:
            a[i] = b[i] * w
    return f

f = make_f(a, b, w, ID)

参见 this analysis哪种类型的变量访问最快。

从算法上讲，没有太多可以做的。 ID 的循环很快，因为它只是增加现有整数的引用计数。整数的散列实际上是即时的，因为 int 的散列是 int 本身。字典查找本身已经高度优化。同样，乘以常数也没有捷径。