python3中利用 timeit 模块计算代码块运行时间
参考文档:有关timeit、repeat方法更深层定义等信息可参考下边博客:
示例代码如下:
import timeit
def get_week():
week_str = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday'] * 100
i = 0
for x in week_str:
if x == 'Monday':
i += 1
a = timeit.timeit(stmt=get_week, number=1000)
print('get_week方法执行1000次耗费时间为:', a)
b = timeit.repeat(stmt=get_week, repeat=3, number=1000)
print('repet3次get_week方法执行1000次耗费时间为:', b)
结果如下:
get_week方法执行1000次耗费时间为: 0.017165
repet3次get_week方法执行1000次耗费时间为: [0.018567099999999996, 0.0199745, 0.0172051]
- timeit.timeit :计算方法(代码块)运行 number 次数所花费时间,返回耗费时间。
- timeit.repeat :计算方法(代码块)重复 repeat 次执行 timeit.timeit 所花费的时间,返回耗时列表。
结束语:通过timeit模块,可在代码块完成之后,对代码块进行压测耗时测试,检测代码在执行多次时间所耗费的时间。