What are some examples of using Numba for performance optimization?

What are some examples of using Numba for performance optimization?

Numba is a just-in-time compiler for Python that can significantly speed up numerical functions. For example, you can use it to optimize loops and array operations.

Can you show me a simple code example?

Sure! Here's a simple example:

from numba import jit

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

This function will run much faster with Numba's JIT compilation.

What types of functions benefit the most from Numba?

Functions that involve heavy numerical computations, such as loops, array manipulations, and mathematical operations, benefit the most from Numba.

Are there any limitations to using Numba?

Yes, Numba works best with NumPy arrays and functions that can be compiled. It may not support all Python features, such as complex data types or certain libraries.