python--timeit模块

来源:互联网 发布:单片机字节变量bdata 编辑:程序博客网 时间:2024/05/18 21:42

测试一段代码的运行时间,在python里面有个很简单的方法,就是使用timeit模块,使用起来超级方便

下面简单介绍一个timeit模块中的函数

主要就是这两个函数:

1, timeit(stmt=’pass’, setup=’pass’, timer=, number=1000000)

   返回:        返回执行stmt这段代码number遍所用的时间,单位为秒,float型   参数:        stmt:要执行的那段代        setup:执行代码的准备工作,不计入时间,一般是import之类的        timer:这个在win32下是time.clock(),linux下是time.time(),默认的,不用管        number:要执行stmt多少遍

2, repeat(stmt=’pass’, setup=’pass’, timer=, repeat=3, number=1000000)

   这个函数比timeit函数多了一个repeat参数而已,表示重复执行timeit这个过程多少遍,返回一个列表,表示执行每遍的时间

当然,为了方便,python还用了一个Timer类,Timer类里面的函数跟上面介绍的两个函数是一样一样的

t1 = timeit.Timer(stmt='pass', setup='pass',timer=<timer function>)t1.timeit(number=1000000)t1.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 timeitimport mathdef myfun():    for i in range(100):        for j in range(2, 10):            math.pow(i, 1/j)n = 100t1 = timeit.timeit(stmt=myfun, number=n)print t1t2 = timeit.repeat(stmt=myfun, number=n, repeat=5)print t2print '-'*110timeitObj = timeit.Timer(stmt=myfun)t3 = timeitObj.timeit(number=n)print t3t4 = timeitObj.repeat(number=n, repeat=5)print t4

输出:

0.02159578478[0.02155279851793634, 0.022432733717750272, 0.022342270091666602, 0.022194384518529814, 0.023163179379425927]---------------------------------------------------------------------------------------------0.0239638787082[0.024934277534064214, 0.022558805366866885, 0.023769157356986453, 0.02344387325468561, 0.022106487236448535][Finished in 0.4s]

*当stmt语句是函数foo()时,应当设置setup=’from main import foo’

def myfun(x):    return 3**xt1 = timeit.timeit(stmt='myfun(100)',setup='from __main__ import myfun',number=100)t2 = timeit.repeat(stmt='myfun(100)',setup='from __main__ import myfun',number=100,repeat=5)print t1print t2#输出:-------------------------------------------------------0.000111956757104[0.00010682406909881228, 0.00010618248309821878, 0.00010586169009792207, 0.00010522010409732857, 0.00015333905414183864]
原创粉丝点击