Python timeit模块

Posted 2020-09-28

測试一段代码的执行时间。在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

为了输出美观，使用了pprint这个有意思的模块