Numba를 사용하여 Python 프로그램에서 수치 계산 속도를 높이는 방법-파이썬 튜토리얼-php.cn

Numba를 사용하여 Python 프로그램에서 수치 계산 속도를 높이는 방법

PHPz

풀어 주다： 2023-08-02 17:37:58

원래의

1474명이 탐색했습니다.

Numba를 사용하여 Python 프로그램의 수치 계산 속도를 높이는 방법

소개:
Python은 수치 계산에 있어 매우 유연하고 사용하기 쉬운 언어입니다. 그러나 Python은 해석된 언어이기 때문에 특히 집중적인 수치 계산 작업에서 상대적으로 느리게 실행됩니다. Python 프로그램의 성능을 향상시키기 위해 일부 최적화 도구와 라이브러리를 사용할 수 있습니다. 매우 강력한 라이브러리 중 하나는 JIT(Just-In-Time) 컴파일을 사용하여 Python 코드 구조를 변경하지 않고 수치 계산 속도를 높이는 Numba입니다. 이 기사에서는 Numba를 사용하여 Python 프로그램의 수치 계산 속도를 높이는 방법을 소개합니다.

Numba 설치:
Numba를 사용하려면 먼저 Numba를 설치해야 합니다. Numba는 pip 패키지 관리자를 사용하여 설치할 수 있습니다:
```
pip install numba
```
로그인 후 복사
기본 사용법:
Numba를 사용하는 가장 쉬운 방법은 데코레이터를 사용하여 가속화해야 하는 기능에 적용하는 것입니다. Numba는 @jit와 @njit라는 두 가지 주요 데코레이터를 지원합니다. @jit 데코레이터를 함수에 적용하고 이를 기계어 코드로 컴파일하여 성능을 향상시킬 수 있습니다. @njit 데코레이터는 Python 인터프리터를 사용하지 않고 함수를 순수한 기계어 코드로 변환하는 @jit(nopython=True)의 단축어입니다. 다음은 간단한 예입니다. @jit和@njit。@jit装饰器可以应用于函数，将其编译为机器码以提高性能。@njit装饰器是@jit(nopython=True)的一个快捷方式，它会将函数转换为不使用Python解释器的纯机器码。下面是一个简单的例子：
```
from numba import jit

@jit
def sum_array(arr):
 total = 0
 for i in range(len(arr)):
     total += arr[i]
 return total

arr = [1, 2, 3, 4, 5]
result = sum_array(arr)
print(result)
```
로그인 후 복사

在上面的例子中，sum_array函数使用@jit装饰器进行了优化。Numba会自动推断函数中变量的类型，并将其编译为机器码。这样，函数的性能会得到大幅提升。

类型推断和类型注解：
为了最大程度地提高性能，Numba需要确切地了解函数和变量的类型。在上面的例子中，Numba可以正确地推断出sum_array函数的类型。然而，在一些情况下，Numba可能无法自动推断类型，这时我们需要使用类型注解来帮助Numba准确地编译函数。下面是一个使用类型注解的例子：
```
from numba import jit

@jit('float64(float64[:])')
def sum_array(arr):
 total = 0
 for i in range(len(arr)):
     total += arr[i]
 return total

arr = [1.0, 2.0, 3.0, 4.0, 5.0]
result = sum_array(arr)
print(result)
```
로그인 후 복사

在上面的例子中，我们通过@jit('float64(float64[:])')注解明确告诉Numbasum_array函数的输入和输出类型。这样，Numba可以更好地优化函数。

并行计算：
Numba还支持并行计算，可以利用多核CPU提高计算性能。要使用并行计算，需要将@jit装饰器的并行参数设置为True：

from numba import njit

@njit(parallel=True)
def parallel_sum(arr):
 total = 0
 for i in range(len(arr)):
     total += arr[i]
 return total

arr = [1, 2, 3, 4, 5]
result = parallel_sum(arr)
print(result)

로그인 후 복사

在上面的例子中，parallel_sum函数通过将@njit(parallel=True)应用于函数上来实现并行计算。这样就可以同时利用多个CPU核心来加速计算。

使用Numba编译生成的代码：
有时候我们可能想要查看Numba编译生成的机器码。可以通过inspect_llvm和inspect_asm函数来查看Numba生成的LLVM代码和汇编代码：

from numba import jit, inspect_llvm, inspect_asm

@jit
def sum_array(arr):
 total = 0
 for i in range(len(arr)):
     total += arr[i]
 return total

arr = [1, 2, 3, 4, 5]
result = sum_array(arr)

print(inspect_llvm(sum_array))  # 查看LLVM代码
print(inspect_asm(sum_array))  # 查看汇编代码

로그인 후 복사

在上面的例子中，我们使用inspect_llvm和inspect_asm函数来查看sum_arrayrrreee

위 예에서 sum_array 함수는 @jit 데코레이터를 사용하여 최적화되었습니다. Numba는 함수의 변수 유형을 자동으로 추론하고 이를 기계어 코드로 컴파일합니다. 이러한 방식으로 기능의 성능이 크게 향상됩니다.

sum_array