timeit模块的使用

2019/10/06 Python Python模块

timeit模块的使用

timeit 主要是为了测试代码运行速度。

它主要有两个方法,即 timeit 和 repeat。

测试一段代码的运行时间,在 python 里面有个很简单的方法,就是使用 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多少遍
  1. 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)

看懂了吧,一样的,使用的时候哪种方便用哪种。

就相当于

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)

Search

    Table of Contents