风火编程--numba使用示例

用numba提升numpy的性能

使用numba可以对numpy的向量运算在时间和空间上进行优化.
在jupyter notebook 和ipython等实时交互中可以通过魔法方法%timeit 和 %memit 进行时间和内存消耗情况的输出.

代码

import time
import numpy as np
import numba as nb
def f_np(i, j):
arr = np.ones((i, j), dtype=np.float64)
np.sum(np.cos(np.log(arr)))

t0 = time.time()
f_np(100000, 2000)
t = time.time() - t0
print(“f_np 耗时:{}”.format(t))

f_nb = nb.jit(f_np)

t0 = time.time()
f_nb(100000, 2000)
t = time.time() - t0
print(“f_nb 耗时:{}”.format(t))