当前位置: 首页 > 工具软件 > Numba > 使用案例 >

numba

双志强
2023-12-01

numba是一款将Python函数编译为代码的JIT编译器,经过numba编译的Python代码(仅限数组运算),其运行速度可以接近C或FORTRAN语言。
Python之所以慢,是因为它是靠CPython编译的,numba的作用是给Python换一种编译器。

装饰器
装饰器本身是一个函数,是函数的函数,目的是增减函数的功能,比如首先定义一个输出当前时间的函数,再定义一个规定时间格式的函数,把后一个函数作用在前一个函数上,就是一个装饰器,作用是用特定格式输出当前时间。

numba的优势

  • 简单,往往只要1行代码就有惊喜。
  • 对循环(loop)有奇效。
  • 兼容常用的科学计算包,如numpy、cmath等。
  • 可以创建ufunc。
  • 会自动调整精度,保证准确性。

使用方法

  • 导入numba及其编译器

    import numba
    from numba import jit
    
  • 传入numba装饰器jit,编写函数

    #传入jit,numba装饰器中的一种
    @jit(nopython=True)
    def function():
        ....
    

参考资料
如何用numba加速python?
numba,让python速度提升百倍

 类似资料: