Python timeit模块
測试一段代码的执行时间。在python里面有个非常easy的方法,就是使用timeit模块。使用起来超级方便
以下简介一个timeit模块中的函数
主要就是这两个函数:
1, timeit(stmt='pass', setup='pass', timer=<defaulttimer>, number=1000000)
返回:
返回运行stmt这段代码number遍所用的时间,单位为秒。float型
參数:
stmt:要运行的那段代码
setup:运行代码的准备工作,不计入时间,通常是import之类的
timer:这个在win32下是time.clock(),linux下是time.time(),默认的,不用管
number:要运行stmt多少遍
2, repeat(stmt='pass', setup='pass', timer=<defaulttimer>, repeat=3, number=1000000)
这个函数比timeit函数多了一个repeat參数而已,表示反复运行timeit这个过程多少遍。返回一个列表,表示运行每遍的时间
当然。为了方便,python还用了一个Timer类。Timer类里面的函数跟上面介绍的两个函数是一样一样的
class timeit.Timer(stmt='pass', setup='pass',timer=<timer function>)
Timer.timeit(number=1000000)
Timer.repeat(repeat=3,number=1000000)
看懂了吧,一样的,使用的时候哪种方便用哪种 O(∩_∩)O
就相当于
timeit(stmt='pass', setup='pass', timer=<defaulttimer>, number=1000000)
= Timer(stmt='pass', setup='pass', timer=<timerfunction>).timeit(number=1000000)
repeat(stmt='pass', setup='pass', timer=<defaulttimer>, repeat=3, number=1000000)
= Timer(stmt='pass', setup='pass', timer=<timerfunction>).repeat(repeat=3, number=1000000)
以下看个简单的样例吧
import timeit import math import pprint def myfun(): for i in range(100): for j in range(2, 10): math.pow(i, 1/j) n = 100 t1 = timeit.timeit(stmt=myfun, number=n) pprint.pprint(t1) t2 = timeit.repeat(stmt=myfun, number=n, repeat=5) pprint.pprint(t2) print() timeitObj = timeit.Timer(stmt=myfun) t3 = timeitObj.timeit(number=n) pprint.pprint(t3) t4 = timeitObj.repeat(number=n, repeat=5) pprint.pprint(t4)
输出结果为:
0.05253329430442725 [0.04811309220489818, 0.04411143942591811, 0.04442468209438866, 0.04363309734529072, 0.0596289993101522] 0.05972538166968161 [0.052578808196427296, 0.047218431876859035, 0.0431364604649378, 0.040855411289408616, 0.04188750572270261] 0.04400300979614258