Timeit模块

Posted 2020-11-16 hanwenlin

对程序的运行时间进行统计的模块timeit

import timeit

timeit 有timeit,repeat方法,Timer类，包含timeit，repeat方法

1.timeit方法

timeit.timeit(stmt=‘add()‘, setup=‘from __main__ import add‘, number=1000)

stmt  运行的程序，测序的函数或者语句

setup 运行的环境  from __main__ import add   即在  if __name__ ==‘__main__‘的环境中

number  测试的次数

 

def add():

　　return sum(list(range(10000)))

 

2. repeat方法  

重复执行测试任务的

all_t = timeit.repeat(stmt=‘add()‘, setup=‘from __main__ import add‘, number=1000,repeat=100)

返回的是每次的测试时间，可以求平均值

sum(all_t)/len(repeatnum)

 

3. Timer() 类

先得到Timer对象，然后使用timeit, repeat方法

t = Timer(stmt=‘add()‘, setup=‘from __main__ import add‘)

t.timeit(number=1500)

t.repeat(number=1500,repeat=1000)