⚡ 简介

Numba是一个开源的JIT编译器，可以将Python代码编译成机器码，大幅提升数值计算性能。

🚀 基本使用

import numpy as np
from numba import jit

@jit(nopython=True)
def go_fast(a):
    trace = 0.0
    for i in range(a.shape[0]):
        trace += np.tanh(a[i, i])
    return a + trace

x = np.arange(10000).reshape(100, 100)
result = go_fast(x)
print(result)

💡 使用要点

1. nopython模式: @jit(nopython=True) 强制编译为机器码
2. 循环友好: Numba对循环有很好的优化
3. NumPy兼容: 支持大部分NumPy函数
4. 首次调用: 第一次调用时会编译，后续调用会更快