diff --git a/2022.03.16_frequently_used_python_package/numba_example.py b/2022.03.16_frequently_used_python_package/numba_example.py new file mode 100644 index 0000000..9f97800 --- /dev/null +++ b/2022.03.16_frequently_used_python_package/numba_example.py @@ -0,0 +1,20 @@ +import numpy as np +import time + +numpy_array = np.arange(0,1e5,1) +times = 1000 + +from numba import jit +from numba import prange +@jit(nopython=True, parallel=True) +def numba_example(numpy_array): + sum = 0 + for i in prange(len(numpy_array)): + sum += numpy_array[i] + return sum + +start = time.time() +for _ in range(times): + result = numba_example(numpy_array) +end = time.time() +print(f'运行时间:{end - start}') \ No newline at end of file