4

我有以下 for 循环来构造值列表：

p = 7
A = []

for i in range(0, 10**p):
    A.append(i**3 + i**2)

为了加快列表的创建，我使用矢量化方法将其创建为 Numpy 数组。这种方法比等效的 for 循环快得多，特别是对于p增加范围的大值。

import numpy as np
from numba import autojit

p = 7
m = np.arange(0, 10**p)
D = np.empty(len(m))
D = m**3 + m**2

为了进一步加快数组的创建速度，我想我会尝试使用 Numba 包。以下是我目前的尝试。

@autojit
def func(a):
    a = np.asarray(a)
    n = np.arange(0, 10**p)
    a = np.append(a, n**3 + n**2)
    return a

e = []
E = func(e)

不幸的是，我没有看到使用 Numba 带来的任何性能提升，它比仅使用 Numpy 的矢量化方法慢近 3 倍。

关于如何使用 Numba 的任何建议？

4

1 回答 1

6

Numba 不会使任意方法调用更快。如果您正在调用图书馆，numba 大部分时间真的无能为力。但是，如果您以不同的方式重新编写内容，您仍然可以获得不错的加速（我使用的是 numba 0.14.0——如果您使用不同的版本、硬件等，您可能会得到不同的结果，尤其是因为 numba正在积极开发中）：

import numpy as np
import numba as nb

def func(a, p):
    a = np.asarray(a)
    n = np.arange(0, 10**p)
    a = np.append(a, n**3 + n**2)
    return a

@nb.jit
def func2(a, p):
    a = np.asarray(a)
    n = np.empty(10**p, dtype=np.float64)
    for k in range(10**p):
        n[k] = k*k*(k + 1)

    return np.append(a, n)

p = 6
e = []
E = func(e, p)
E2 = func2(e, p)
print np.allclose(E, E2)

和时间：

In [51]:

%timeit func(e, p)
10 loops, best of 3: 42.9 ms per loop
In [52]:

%timeit func2(e, p)
100 loops, best of 3: 3.09 ms per loop

另外，p=7您需要注意数值精度。

numba 的关键是展开循环并且只进行 numba 支持的“原始”算术调用

于 2014-10-14T20:29:19.740 回答