代码性能测试(timeit)

测试代码性能timeit

  • 在测试代码的性能时,我常用time.time()简单判断一个函数执行的大约时间

    import time
    
    # 记录开始时间
    start_time = time.time()
    
    # 计算从10000的阶乘
    total = 1
    for i in range(1, 10000):
     total = total * i
    
    # 记录结束时间
    end_time = time.time()
    
    # 计算和打印用时
    used_time = end_time - start_time
    print(f'共用时:{used_time}')
    
    #time()是time模块中的一个函数,它返回自从1970年0点到现在一共经历的秒数
    
  • 但利用time.time()每次时间可能会有差异。如果主要目的是为了测试一个函数的性能,还有更好的办法——timeit

import timeit
s = """
L = []
for n in range(10):
    L.append(2 ** n)"""
t = timeit.timeit(s)
print(t)
#在timeit(),如果要测试多行代码的性能,可以通过三括号把代码包起来

#timeit重复执行上面的代码100万次(默认值),然后打印出总用时

t = timeit.timeit(s,number=100)#可以调整执行次数
posted @ 2023-01-17 11:04  小杨的冥想课  阅读(99)  评论(0编辑  收藏  举报