Numba 是一个开源的即时编译器,可以将 Python 代码转换为机器码,从而加速数值计算。

安装

要安装 Numba,请使用以下命令:

pip install numba

快速开始

以下是一个使用 Numba 加速 NumPy 数组操作的例子:

from numba import jit
import numpy as np

@jit(nopython=True)
def add(a, b):
    return a + b

a = np.random.rand(1000, 1000)
b = np.random.rand(1000, 1000)

result = add(a, b)

高级功能

Numba 提供了多种高级功能,例如:

  • 类型注解:提高性能的同时,还可以利用类型信息。
  • 生成器支持:可以优化生成器函数。
  • 动态函数:允许在运行时动态生成函数。

资源

更多信息和文档,请访问 Numba 官方网站

示例图片

Numba 示例