Python使用timeit模块计算某一组代码块的平均耗时

timeit模块：该模块可以用来测试一段python代码的执行速度/时长

Timer类：该类是timeit模块中专门用于测量python代码的执行速度/时长的。原型为class timeit.Timer(stmt='pass',setup='pass')。

stmt参数：表示即将进行测试的代码语句。

setup: 运行代码块语句时所需要的设置。

timeit函数：timeit.Timer.timeit(number=100000),该函数返回代码块语句执行number此的平均耗时。

from timeit import Timer

def test01():
    alist = []
    for i in range(1000):
        alist.append(i)
    return alist

def test02():
    alist = []
    for i in range(1000):
        alist += [i]
    return alist

def test03():
    alist = [i for i in range(1000)]

def test04():
    alist = list(range(1000))
    return alist

if __name__ == '__main__':
    t1 = Timer('test01()',setup='from __main__ import test01')
    second_1 = t1.timeit(1000)
    print('函数1',second_1)
    
    t2 = Timer('test02()',setup='from __main__ import test02')
    second_2 = t2.timeit(1000)
    print('函数2',second_2)
    
    t3 = Timer('test03()',setup='from __main__ import test03')
    second_3 = t3.timeit(1000)
    print('函数3',second_3)
    
    t4 = Timer('test04()',setup='from __main__ import test04')
    second_4 = t4.timeit(1000)
    print('函数4',second_4)

运行结果：

函数1 0.05753669999990052
函数2 0.07849220000002788
函数3 0.025955500000009124
函数4 0.01134290000004512