From 7b07bd90bc637d4218379aabd447afc87f3d4ef3 Mon Sep 17 00:00:00 2001 From: guanjihuan Date: Sun, 23 Mar 2025 22:16:38 +0800 Subject: [PATCH] Create numba_example.py --- .../numba_example.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 2022.03.16_frequently_used_python_package/numba_example.py 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