From f9b54d7b1ccfd31f24690eaf5f5d8e128d97b0ab Mon Sep 17 00:00:00 2001 From: guanjihuan Date: Sun, 6 Mar 2022 16:16:43 +0800 Subject: [PATCH] Create numba_time.py --- .../2022.03.06_numba_time/numba_time.py | 42 +++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 language_learning/2022.03.06_numba_time/numba_time.py diff --git a/language_learning/2022.03.06_numba_time/numba_time.py b/language_learning/2022.03.06_numba_time/numba_time.py new file mode 100644 index 0000000..c23c5a8 --- /dev/null +++ b/language_learning/2022.03.06_numba_time/numba_time.py @@ -0,0 +1,42 @@ +import numpy as np +from numba import jit +import time + +def for_sum(numpy_array): + sum = 0 + for number in numpy_array: + sum += number + return sum + +@jit +def numba_for_sum(numpy_array): + sum = 0 + for number in numpy_array: + sum += number + return sum + +numpy_array = np.arange(0,1e8,1) + +start = time.time() +result = sum(numpy_array) +end = time.time() +print('\nresult:', result) +print('python中sum()函数求和时间:\n', end - start) + +start = time.time() +result = np.sum(numpy_array) +end = time.time() +print('\nresult:', result) +print('numpy.sum()函数求和时间:\n', end - start) + +start = time.time() +result = for_sum(numpy_array) +end = time.time() +print('\nresult:', result) +print('for循环求和numpy数组的时间:\n', end - start) + +start = time.time() +result = numba_for_sum(numpy_array) +end = time.time() +print('\nresult:', result) +print('numba加速for循环求和numpy数组的时间:\n', end - start, '\n') \ No newline at end of file