Comment utiliser Numba pour accélérer les calculs numériques dans les programmes Python

Comment utiliser Numba pour accélérer les calculs numériques des programmes Python

Introduction :
Python est un langage très flexible et facile à utiliser en matière de calculs numériques. Cependant, Python étant un langage interprété, il s’exécute relativement lentement, en particulier dans les tâches de calcul numérique intensives. Afin d'améliorer les performances des programmes Python, nous pouvons utiliser certains outils et bibliothèques d'optimisation. Numba est une bibliothèque très puissante qui utilise la compilation juste à temps pour accélérer les calculs numériques sans modifier la structure du code Python. Cet article explique comment utiliser Numba pour accélérer le calcul numérique des programmes Python.

1. Installer Numba :
   Pour commencer à utiliser Numba, vous devez d'abord l'installer. Numba peut être installé en utilisant le gestionnaire de paquets pip :

   pip install numba

   Copier après la connexion

2. Utilisation de base :
   La façon la plus simple d'utiliser Numba est d'utiliser un décorateur pour l'appliquer à la fonction qui doit être accélérée. Numba prend en charge deux décorateurs principaux : @jit et @njit. Le décorateur @jit peut être appliqué aux fonctions, en les compilant en code machine pour améliorer les performances. Le décorateur @njit est un raccourci pour @jit(nopython=True), qui convertit une fonction en code machine pur sans utiliser l'interpréteur Python. Voici un exemple simple : @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)

   Copier après la connexion

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

3. 类型推断和类型注解：
   为了最大程度地提高性能，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)

   Copier après la connexion

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

4. 并行计算：
   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)

   Copier après la connexion

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

5. 使用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))  # 查看汇编代码

   Copier après la connexion

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

Dans l'exemple ci-dessus, la fonction sum_array est optimisée à l'aide du décorateur @jit. Numba déduit automatiquement les types de variables dans les fonctions et les compile en code machine. De cette façon, les performances de la fonction seront grandement améliorées.

Inférence de type et annotations de type :
Pour maximiser les performances, Numba doit connaître exactement les types de fonctions et de variables. Dans l'exemple ci-dessus, Numba peut déduire correctement le type de la fonction sum_array. Cependant, dans certains cas, Numba peut ne pas être en mesure de déduire automatiquement le type. Dans ce cas, nous devons utiliser des annotations de type pour aider Numba à compiler la fonction avec précision. Voici un exemple d'utilisation des annotations de type :
1. rrreee
2. Dans l'exemple ci-dessus, nous indiquons explicitement à Numbasum_array via l'annotation <code>@jit('float64(float64[:])') Les types d'entrée et de sortie de la fonction. De cette façon, Numba peut mieux optimiser les fonctions.
Calcul parallèle : 🎜Numba prend également en charge le calcul parallèle, qui peut utiliser des processeurs multicœurs pour améliorer les performances informatiques. Pour utiliser le calcul parallèle, vous devez définir le paramètre parallèle du décorateur @jit sur True : 🎜rrreee🎜🎜🎜Dans l'exemple ci-dessus, parallel_sum code >Functions implémente le calcul parallèle en appliquant <code>@njit(parallel=True) à la fonction. Cela permet d'utiliser plusieurs cœurs de processeur simultanément pour accélérer les calculs. 🎜
🎜🎜Code généré à l'aide de la compilation Numba : 🎜Parfois, nous pouvons souhaiter afficher le code machine généré par la compilation Numba. Le code LLVM et le code assembleur générés par Numba peuvent être visualisés via les fonctions inspect_llvm et inspect_asm : 🎜rrreee🎜🎜🎜Dans l'exemple ci-dessus, nous utilisons inspect_llvm code> et <code>inspect_asm pour afficher le code LLVM et le code assembleur de la fonction sum_array. 🎜🎜Conclusion : 🎜L'utilisation de Numba peut améliorer considérablement les performances de calcul numérique des programmes Python. En ajoutant simplement un décorateur à la fonction qui doit être accélérée, nous pouvons profiter de la fonctionnalité de compilation juste à temps de Numba pour compiler le code Python en code machine efficace. De plus, Numba prend également en charge l'inférence de type, les annotations de type et le calcul parallèle, offrant davantage d'options d'optimisation. En utilisant Numba, nous pouvons mieux tirer parti de la simplicité et de la flexibilité de Python tout en atteignant des performances de langage de programmation quasi natives. 🎜🎜Références : 🎜🎜🎜https://numba.pydata.org/🎜🎜https://numba.pydata.org/numba-doc/latest/user/jit.html🎜🎜https://numba.pydata . org/numba-doc/latest/user/examples.html🎜🎜

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!