timeit的功能和用法<a id="sec-1" name="sec-1"></a>
timeit 模塊提供了測(cè)試一小段代碼運(yùn)行時(shí)間的功能愧口。我前面有一篇文章用它來測(cè)試定義 __slots__ 對(duì)對(duì)象訪問速度的提升情況，參見這里 。
官方文檔 上有下面這樣的使用例子：
# 從命令行調(diào)用
python -m timeit '"-".join([str(n) for n in range(100)])'
# 從REPL調(diào)用
>>> import timeit
>>> timeit.timeit('"-".join([str(n) for n in range(100)])', number=10000)
0.7288308143615723
使用 timeit 的時(shí)候可以直接調(diào)用其定義好的 timeit.timeit , timeit.repeat , timeit.default_timer 方法哗总，或者定義一個(gè)類 timeit.Timer ,使用本身的方法舔涎。
# API
timeit.timeit(stmt='pass', setup='pass', timer=<default timer>, number=1000000)
timeit.repeat(stmt='pass', setup='pass', timer=<default timer>, repeat=3, number=1000000)
class timeit.Timer(stmt='pass', setup='pass', timer=<timer function>)?
-
timeit()接受的第一個(gè)參數(shù)stmt為要計(jì)算時(shí)間的表達(dá)式，第二個(gè)參數(shù)setup為初始化的表達(dá)式获列。 - timer參數(shù)為計(jì)時(shí)器谷市，在不同平臺(tái)默認(rèn)不一樣，參照這里 击孩，但不管是Unix還是Windows測(cè)得的時(shí)間都是系統(tǒng)經(jīng)過的時(shí)間迫悠，所以是有可能被正在運(yùn)行的其它程序影響的。
- number參數(shù)是表達(dá)式執(zhí)行的次數(shù)巩梢，所以這里返回的時(shí)間是執(zhí)行1000000次總和的時(shí)間创泄。
-
timeit.repeat()中的repeat參數(shù)是測(cè)試重復(fù)執(zhí)行的次數(shù)，相當(dāng)于調(diào)用number次的timeit(),number默認(rèn)為3括蝠，返回3次運(yùn)行時(shí)間的列表鞠抑，通常可以取min()來評(píng)估程序的執(zhí)行時(shí)間忌警。 -
timeit.repeat()中除了最小值其他的值得參考意義可能不大搁拙，因?yàn)橛锌赡苁鞘艿较到y(tǒng)中其它程序影響導(dǎo)致的 - 注意python版本不同導(dǎo)致的接口參數(shù)不同，這里默認(rèn)采用python2.7版本
接下來看一些復(fù)雜點(diǎn)的示例
# 多行表達(dá)式的調(diào)用
>>> import timeit
>>> attribute is missing
>>> s = """\\
... try:
... str.__nonzero__
... except AttributeError:
... pass
... """
>>> timeit.timeit(stmt=s, number=100000)
0.9138244460009446
# 測(cè)試函數(shù)調(diào)用時(shí)間
def test():
"""Stupid test function"""
L = []
for i in range(100):
L.append(i)
if __name__ == '__main__':
import timeit
print(timeit.timeit("test()", setup="from __main__ import test"))
其它<a id="sec-2" name="sec-2"></a>
當(dāng)我在網(wǎng)上瀏覽timeit相關(guān)資源時(shí)看到了這篇文章法绵， 上面提到了通過上下文管理器(context manager)實(shí)現(xiàn)一個(gè)計(jì)時(shí)器箕速，利用了變量的生命周期相關(guān)特性(#TODO)，雖然就像作者所說礼烈，可以加上其它的特性比如循環(huán)次數(shù)等以及更優(yōu)雅的實(shí)現(xiàn)一個(gè)計(jì)時(shí)器弧满，代碼如下,可供參考：
# Source code: https://gist.github.com/pengmeng/78a25663c20ab8890b81
__author__ = 'mengpeng'
import time
class timeme(object):
__unitfactor = {'s': 1,
'ms': 1000,
'us': 1000000}
def __init__(self, unit='s', precision=4):
self.start = None
self.end = None
self.total = 0
self.unit = unit
self.precision = precision
def __enter__(self):
if self.unit not in timeme.__unitfactor:
raise KeyError('Unsupported time unit.')
self.start = time.time()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.end = time.time()
self.total = (self.end - self.start) * timeme.__unitfactor[self.unit]
self.total = round(self.total, self.precision)
def __str__(self):
return 'Running time is {0}{1}'.format(self.total, self.unit)
運(yùn)行示例:
from timeme import timeme
with timeme('ms', 6) as t:
result = sum(range(100000))
print(t) # Running time is 5.2948ms
print(t.total) # 5.2948
