news 2026/8/8 0:34:28

numba,让你的Python飞起来!

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
numba,让你的Python飞起来!

python由于它动态解释性语言的特性,跑起代码来相比java、c++要慢很多,尤其在做科学计算的时候,十亿百亿级别的运算,让python的这种劣势更加凸显。

办法永远比困难多,numba就是解决python慢的一大利器,可以让python的运行速度提升上百倍!

什么是numba?

numba是一款可以将python函数编译为机器代码的JIT编译器,经过numba编译的python代码(仅限数组运算),其运行速度可以接近C或FORTRAN语言。

python之所以慢,是因为它是靠CPython编译的,numba的作用是给python换一种编译器。

python、c、numba三种编译器速度对比

使用numba非常简单,只需要将numba装饰器应用到python函数中,无需改动原本的python代码,numba会自动完成剩余的工作。

import numpy as np

import numbafrom numba import jit@jit(nopython=True) # jit,numba装饰器中的一种def go_fast(a): # 首次调用时,函数被编译为机器代码trace = 0# 假设输入变量是numpy数组for i in range(a.shape[0]): # Numba 擅长处理循环trace += np.tanh(a[i, i])return a + trace

以上代码是一个python函数,用以计算numpy数组各个数值的双曲正切值,我们使用了numba装饰器,它将这个python函数编译为等效的机器代码,可以大大减少运行时间。

numba适合科学计算

numpy是为面向numpy数组的计算任务而设计的。

在面向数组的计算任务中,数据并行性对于像GPU这样的加速器是很自然的。Numba了解NumPy数组类型,并使用它们生成高效的编译代码,用于在GPU或多核CPU上执行。特殊装饰器还可以创建函数,像numpy函数那样在numpy数组上广播。

什么情况下使用numba呢?

  • 使用numpy数组做大量科学计算时

  • 使用for循环时

学习使用numba

第一步:导入numpy、numba及其编译器

import numpy as np

import numbafrom numba import jit

第二步:传入numba装饰器jit,编写函数

# 传入jit,numba装饰器中的一种@jit(nopython=True)def go_fast(a): # 首次调用时,函数被编译为机器代码trace = 0# 假设输入变量是numpy数组for i in range(a.shape[0]): # Numba 擅长处理循环trace += np.tanh(a[i, i]) # numba喜欢numpy函数return a + trace # numba喜欢numpy广播

nopython = True选项要求完全编译该函数(以便完全删除Python解释器调用),否则会引发异常。这些异常通常表示函数中需要修改的位置,以实现优于Python的性能。强烈建议您始终使用nopython = True。

第三步:给函数传递实参

# 因为函数要求传入的参数是nunpy数组

x = np.arange(100).reshape(10, 10)# 执行函数go_fast(x)

第四步:经numba加速的函数执行时间

% timeit go_fast(x)
输出:
3.63 µs ± 156 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

第五步:不经numba加速的函数执行时间

def go_fast(a): # 首次调用时,函数被编译为机器代码trace = 0# 假设输入变量是numpy数组for i in range(a.shape[0]): # Numba 擅长处理循环trace += np.tanh(a[i, i]) # numba喜欢numpy函数return a + trace # numba喜欢numpy广播x = np.arange(100).reshape(10, 10)%timeit go_fast(x)

输出:

136 µs ± 1.09 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

结论:

在numba加速下,代码执行时间为3.63微秒/循环。不经过numba加速,代码执行时间为136微秒/循环,两者相比,前者快了40倍。

numba让python飞起来

前面已经对比了numba使用前后,python代码速度提升了40倍,但这还不是最快的。

这次,我们不使用numpy数组,仅用for循环,看看nunba对for循环到底有多钟爱!​​​​​​​

# 不使用numba的情况def t():x = 0for i in np.arange(5000):x += ireturn x%timeit(t())

输出:

408 µs ± 9.73 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
# 使用numba的情况@jit(nopython=True)def t():x = 0for i in np.arange(5000):x += ireturn x%timeit(t())

输出:

1.57 µs ± 53.8 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

使用numba前后分别是408微秒/循环、1.57微秒/循环,速度整整提升了200多倍!

结语

numba对python代码运行速度有巨大的提升,这极大的促进了大数据时代的python数据分析能力,对数据科学工作者来说,这真是一个lucky tool !

当然numba不会对numpy和for循环以外的python代码有很大帮助,你不要指望numba可以帮你加快从数据库取数,这点它真的做不到哈。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/7 10:47:54

攻克OpenVLA模型微调中的归一化陷阱:实战指南与解决方案

在机器人视觉语言动作模型的实际应用中,OpenVLA模型微调后的推理阶段常常隐藏着一个技术陷阱——归一化问题。这不仅是理论概念,更是影响模型部署成功的关键因素。本文将带你深入理解这一机制,并提供切实可行的解决方案。 【免费下载链接】op…

作者头像 李华
网站建设 2026/8/7 17:06:30

java计算机毕业设计人力资源管理系统的设计与实现 基于SpringBoot的企业人事综合管理平台研发 面向中小企业的智能人事与考勤一体化系统

计算机毕业设计人力资源管理系统的设计与实现6c3x69(配套有源码 程序 mysql数据库 论文) 本套源码可以在文本联xi,先看具体系统功能演示视频领取,可分享源码参考。当“人力”成为企业最昂贵的资产,传统Excel纸质审批的模式就像老牛…

作者头像 李华
网站建设 2026/8/8 10:40:49

Rust语言学习笔记

本文假设你已经学过cpp语言咧变量部分变量可以自动推断类型。变量默认不可改,可改变量要用let mut 进行声明,同名可以进行遮蔽,不限类型。变量类型用 :i32这种在let或者let mut 后进行声明。元组VS数组:元组是任意变量…

作者头像 李华
网站建设 2026/8/7 7:19:29

算法题 设计哈希映射

设计哈希映射 问题描述 不使用任何内建的哈希表库设计一个哈希映射(HashMap)。 实现 MyHashMap 类: void put(int key, int value) 向哈希映射中插入键值对 (key, value)。如果键已经存在,更新对应的值。int get(int key) 返回特定…

作者头像 李华
网站建设 2026/8/7 15:30:46

PaperXie 如何重塑开题报告的“研究可行性蓝图”属性:一种面向“从构想到落地”闭环构建的智能协作框架——一位研究生的真实实践记录

paperxie-AI官网免费论文查重复率AIGC检测/开题报告/文献综述/论文初稿 https://www.paperxie.cn/ai/openingReporthttps://www.paperxie.cn/ai/openingReport 在研究生培养体系中,开题报告常被视为一项形式化的行政程序。然而,从学术本质来看&#xf…

作者头像 李华
网站建设 2026/8/7 23:53:30

单片机stm32系列——PWM配置应用LED呼吸灯,搭配stm32cubeMX+keil5

1.配置TIM3,设置为内部时钟源,使用CH1通道,会在固定输出PA6引脚口2.打开TIM3中断向量管理器NVIC的使能3.生成代码裸机开发4.打开keil5工程,发现在多了一个tim.c文件,里面就配置了TIM3配置及PWM的配置5.主函数前启动PWM…

作者头像 李华