python--timeit模块

测试一段代码的运行时间，在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=)
t1.repeat(repeat=,number=)
           

看懂了吧，一样的，使用的时候哪种方便用哪种 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
def myfun():
    for i in range():
        for j in range(, ):
            math.pow(i, /j)

n = 

t1 = timeit.timeit(stmt=myfun, number=n)
print t1
t2 = timeit.repeat(stmt=myfun, number=n, repeat=)
print t2

print '-'*

timeitObj = timeit.Timer(stmt=myfun)
t3 = timeitObj.timeit(number=n)
print t3
t4 = timeitObj.repeat(number=n, repeat=)
print t4
           

输出：

[, , , , ]
---------------------------------------------------------------------------------------------

[, , , , ]
[Finished in s]
           

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

def myfun(x):
    return **x
t1 = timeit.timeit(stmt='myfun(100)',setup='from __main__ import myfun',number=)
t2 = timeit.repeat(stmt='myfun(100)',setup='from __main__ import myfun',number=,repeat=)
print t1
print t2
#输出：
-------------------------------------------------------

[, , , , ]