numba的使用比较简单,它不需要更换python的解释器,只需要将numba的装饰器写在python方法上,当这个带有numba装饰器的方法被调用时,就会被just-in-time即时编译为机器代码,然后执行。使用numba之前,我们需要安装这个库
pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com numba
案例:找出2000000以内的质数为例,分别计算下需要花费多长的时间
未使用numba之前:
import math
import time
from numba import njit
@njit(fastmath=True, cache=True)
def is_prime(num):
if num == 2:
return True
if num <= 1 or not num % 2:
return False
for i in range(3, int(math.sqrt(num)) + 1, 2