文 | 菊子皮(转载请注明出处)

关注公众号：

AIAS编程有道

环境：Python 3.7，VS Code

timeit模块使用

有时候需要测试创建某种变量而花费的时间、有时候需要测试某个函数的运行的时间，对于简单的Python程序使用Python自带timeit模块是不错的选择。

这里也只介绍经常使用的几个方法，更多内容可参考官方文档。Python3中的timeit模块可以用来测试小段代码的运行时间。其中主要通过两个函数来实现：timeit和repeat。

timeit. timeit(

stmt=‘pass’

,

setup=‘pass’

,

timer=

,

number=1000000

,

globals=None

)

使用给定语句、

setup

代码和

timer

函数创建一个

Timer

实例，并执行

number

次其

timeit()

方法。可选的

globals

参数指定用于执行代码的命名空间。

timeit.repeat(

stmt=‘pass’

,

setup=‘pass’

,

timer=

,

repeat=5

,

number=1000000

,

globals=None

)

使用给定语句、

setup

代码和

timer

函数创建一个

Timer

实例，并使用给定的

repeat

计数和

number

执行运行其

repeat()

方法。可选的

globals

参数指定用于执行代码的命名空间。

参数说明：

stmt：用于传入要测试时间的代码，可以直接接受字符串的表达式，也可以接受单个变量，也可以接受函数。传入函数时要把函数申明在当前文件中，然后在 stmt = ‘func()’ 执行函数，然后使用 setup = ‘from

main

import func’

setup：传入stmt的运行环境，比如stmt中使用到的参数、变量，要导入的模块等。可以写一行语句，也可以写多行语句，写多行语句时要用分号；隔开语句。

number：要测试的代码的运行次数，默认100000次，对于耗时的代码，运行太多次会比较慢，此时建议自己修改一下运行次数

repeat：指测试要重复几次，每次的结果构成列表返回，默认5次。

使用案例：

# <1>

print(timeit.timeit(stmt='list(i**2 for i in normal_list)', setup='normal_list=range(10000)', number=10))

# 0.0857056 结果也会与计算机性能有关

# <2>

print(timeit.repeat(stmt='list(i**2 for i in normal_list)', setup='normal_list=range(10000)', repeat=2, number=10))

# [0.08216990000000002, 0.055868999999999974]

# <3> setup 为复合语句

print(timeit.timeit(stmt='list(i**2 for i in normal_list)', setup='a=10000;normal_list=range(a)', number=10))

# 0.0660696

# <4>

print(timeit.repeat(stmt='list(i**2 for i in normal_list)', setup='a=10000;normal_list=range(a)', repeat=2, number=10))

# [0.06270019999999998, 0.054296299999999964]

# <5>

def func():

normal_list = range(10000)

L = [i**2 for i in normal_list]

# stmt为函数

print(timeit.timeit("func()", setup="from __main__ import func", number=10))

# 0.056916100000000025

# <6>

print(timeit.repeat("func()", setup="from __main__ import func", repeat=3, number=10))

# [0.057099300000000075, 0.05018290000000003, 0.04987839999999999]

程序说明：

<1> 运行语句

list(i**2 for i in normal_list)

其中语句的参数为

normal_list=range(10000)

一个10次所使用的时间。

<2> 在<1>的基础上，重复运行2次，每次运行的时间都放在一个列表中

<3> 提供参数的语句可以是多个

<4> 在<3>的基础上，重复运行2次，每次运行的时间都放在一个列表中

<5> 使用自定义函数情况

<6> 在<5>的基础上，重复运行3次，每次运行的时间都放在一个列表中