这个库，让我的Python从”拖拉机”变”法拉利”！

Python慢，这事儿地球人都知道。写个循环跑几百万次，泡杯咖啡回来可能还没跑完。但你有没有想过，不改变Python的写法，只加一行代码，速度就能翻几十倍甚至上百倍？今天聊的这玩意儿叫Numba，用过之后你会怀疑之前写的是不是假Python。

Python为啥慢

先搞清楚一个问题：Python到底慢在哪？

Python是解释型语言，代码一行一行地被解释器”翻译”成机器能懂的指令。这个翻译过程本身就需要时间。更要命的是，Python的变量是动态类型的——你可以把一个变量先赋值成整数，下一秒又变成字符串。这种灵活性很爽，但解释器得时刻提心吊胆地检查”这货现在到底是啥类型”，开销就大了。

C语言为啥快？因为它是编译型语言，代码提前编译成机器码，运行时直接执行，没有中间商赚差价。

那能不能把Python代码也编译一下？能，这就是Numba干的事。

Numba是个啥

Numba是一个即时编译器（JIT，Just-In-Time Compiler）。你只需要在函数上面加一个装饰器，Numba就会在函数第一次被调用时，把它编译成机器码缓存起来。后续再调用这个函数，直接跑编译好的机器码，快得飞起。

装它很简单：

pipinstallnumba

来看个最基础的例子：

fromnumbaimport jit
importtime

@jit(nopython=True)
defsum_numbers(n):
    total = 0
    for i in range(n):
        total += i
    return total

# 第一次调用会触发编译，稍慢
start = time.time()
result = sum_numbers(100000000)
print(f"结果: {result}, 耗时: {time.time()-start:.4f}秒")

![](https://imgs.cheemao.com/images/2026/01/07/210c010db9354c3ead77c75c24492a65.jpg)

# 第二次调用直接用缓存的机器码
start = time.time()
result = sum_numbers(100000000)
print(f"结果: {result}, 耗时: {time.time()-start:.4f}秒")

跑一下你会发现，第一次调用大概零点几秒（包含编译时间），第二次调用可能只要零点零几秒。如果把@jit那行去掉，用纯Python跑，得好几秒。

这差距，就是”拖拉机”和”法拉利”的区别。

nopython模式必须开

注意上面代码里@jit(nopython=True)这个参数。nopython=True的意思是告诉Numba：”你必须完全编译这段代码，不许偷偷回退到Python解释器”。

如果不加这个参数，Numba遇到它搞不定的代码会悄悄用Python解释器来跑，速度就没那么快了。建议养成习惯，永远加上nopython=True。懒的话可以用它的简写形式@njit：

fromnumbaimport njit

@njit
defmy_fast_function(x):
    return x * 2

效果一样，写起来更省事。

温馨提示：Numba不是万能的。它主要加速数值计算，对字符串操作、字典、列表推导式这些支持得不太好。如果你的函数里有这类操作，编译可能会失败。遇到报错别慌，看看是不是用了Numba不支持的语法。

配合NumPy更香

Numba和NumPy是绝配。NumPy数组在Numba里可以直接用，而且速度提升更明显。

举个例子，计算两个大数组的逐元素乘积再求和：

importnumpyasnp
fromnumbaimport njit

![](https://imgs.cheemao.com/images/2026/01/07/07febefcf4bb4664a7ab73de3a787bf3.jpg)

@njit
defdot_product(a, b):
    result = 0.0
    for i in range(len(a)):
        result += a[i] * b[i]
    return result

# 生成测试数据
a = np.random.rand(10000000)
b = np.random.rand(10000000)

# 预热（触发编译）
_ = dot_product(a, b)

# 计时
importtime
start = time.time()
result = dot_product(a, b)
print(f"Numba耗时: {time.time()-start:.4f}秒")

# 对比纯NumPy
start = time.time()
result_np = np.dot(a, b)
print(f"NumPy耗时: {time.time()-start:.4f}秒")

这个例子里Numba和NumPy的np.dot速度差不多，因为NumPy底层本身就是C写的。但如果是更复杂的、NumPy没有现成函数的计算逻辑，Numba的优势就体现出来了——你可以用Python语法写循环，性能却接近C。

并行加速：榨干CPU每一个核

现代CPU都是多核的，光用一个核跑未免太浪费。Numba提供了prange来实现并行循环：

fromnumbaimport njit, prange
importnumpyasnp

@njit(parallel=True)
defparallel_sum(arr):
    total = 0.0
    for i in prange(len(arr)):
        total += arr[i]
    return total

data = np.random.rand(100000000)

# 预热
_ = parallel_sum(data)

importtime
start = time.time()
result = parallel_sum(data)
print(f"并行计算结果: {result:.2f}, 耗时: {time.time()-start:.4f}秒")

把range换成prange，再加上parallel=True参数，Numba就会自动把循环分配到多个CPU核心上跑。核心越多，速度越快。

温馨提示：不是所有循环都能并行化。如果循环的每一次迭代依赖前一次的结果（比如斐波那契数列），强行并行会出问题。Numba会尽量检测这种情况，但有时候也会翻车。用之前想清楚你的循环逻辑能不能并行。

还有更狠的：Codon

如果Numba是”涡轮增压”，那Codon就是”换发动机”。

Codon是MIT搞出来的一个Python编译器，它不是JIT，而是提前编译（AOT），直接把Python代码编译成可执行文件。跑起来性能跟C++一个级别，甚至更快。

但Codon有个大问题：它不支持大部分Python第三方库。NumPy、Pandas这些都用不了。所以它更适合从零开始写纯计算逻辑的场景，不太适合在现有项目里用。

对于大多数人来说，Numba才是那个”既能跑得快，又不用改太多代码”的甜蜜点。

什么时候该用Numba

并不是所有代码都值得用Numba优化。记住这几点：

适合用的场景：大量数值计算、循环次数多、计算密集型任务。比如科学计算、金融建模、图像处理里的像素级操作。

不太适合的场景：IO密集型任务（读写文件、网络请求）、大量字符串处理、频繁调用Python内置库的代码。

优化之前先用cProfile或者line_profiler找到真正的性能瓶颈。别一上来就给所有函数加@njit，那样编译时间反而拖慢整体速度。找准那个占用时间最多的函数，对它下手，效果立竿见影。

Numba这玩意儿，说白了就是给Python插上了编译的翅膀。语法还是熟悉的Python语法，但背地里已经变成机器码在狂飙了。下次遇到Python跑得慢想砸键盘的时侯，先试试加个@njit——也许问题就这么解决了。