Bootstrap

python加速模块:numba

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
;