@numba.jit(nopython=True)

yuyuyue249

已于 2024-07-20 12:55:56 修改

阅读量446

点赞数 5

文章标签： python

于 2024-07-11 14:08:49 首次发布

本文链接：https://blog.youkuaiyun.com/yuyuyue249/article/details/140350544

版权

import time
import numpy as np
def cal_sum(a):
    result = 0
    for i in range(a.shape[0]):
        for j in range(a.shape[1]):
            result += a[i, j]
    return result

start = time.perf_counter()
a = np.random.random((5000, 5000))
result = cal_sum(a)
end = time.perf_counter()
print("原始代码耗时：{}s".format((end - start)))

#OUT:
#原始代码耗时：5.725140199996531s
记得点赞哦 ：)
















import numba
import time
import numpy as np

@numba.jit(nopython=True)
def cal_sum(a):
    result = 0
    for i in range(a.shape[0]):
        for j in range(a.shape[1]):
            result += a[i][j]
    return result


start = time.perf_counter()
a = np.random.random((5000, 5000))
result = cal_sum(a)
end = time.perf_counter()
print(f"加速后耗时：{end - start}s")

# OUT:
# 加速后耗时：0.2892118000017945s