其實平時使用測試應(yīng)用運行時間的情況 細算一下還真的很少。很久沒有做性能優(yōu)化的工作，不管是cProfile還是timeit模塊都已經(jīng)生疏了很久沒有使用，我在以前的文章里面有提到過cPfile的性能測試使用，但是一直沒有使用過這個更輕量級的運行時間測量庫進行過仔細實踐總結(jié)，今天就來總結(jié)一下。

從最簡單的例子開始，比如我們想測試一個列表推導(dǎo)式究竟要比正常寫for快多少。

import timeitfoooo = """ sum = [] for i in range(1000):sum.append(i) """print (timeit.timeit(stmt="[i for i in range(1000)]", number=100000)) print (timeit.timeit(stmt=foooo, number=100000))

輸出：

3.79257702827 9.0510661602

不難看出，使用列表推導(dǎo)式要比正常使用list追加元素 通過10w次循環(huán)會快上近5.3秒左右，速度快上近三倍。

timeit 模塊抽象出了；兩個可以直接使用的方法，包了一層可以讓我們不用關(guān)心內(nèi)部的實現(xiàn)，下面看一下模塊里面的代碼：

def timeit(stmt="pass", setup="pass", timer=default_timer,number=default_number):"""Convenience function to create Timer object and call timeit method."""return Timer(stmt, setup, timer).timeit(number)def repeat(stmt="pass", setup="pass", timer=default_timer,repeat=default_repeat, number=default_number):"""Convenience function to create Timer object and call repeat method."""return Timer(stmt, setup, timer).repeat(repeat, number)

可以看到這兩個方法都是對Timer類包了一層這幾個參數(shù)：

stmt: 這個參數(shù)就是statement,可以把要進行計算時間的代碼放在里面。他可以直接接受字符串的表達式，也可以接受單個變量，也可以接受函數(shù)。

• setup: 這個參數(shù)可以將stmt的環(huán)境傳進去。比如各種import和參數(shù)什么的。

• timer: 這個參數(shù)一般使用不到，具體使用可以參看文檔。

Timer類下面還有repeat和timeit方法 使用也非常方便就是 timeit.timeit 和 timeit.repeat。

一個就是上面例子的timeit，一個就是repeat 其實repeat就比timeit多了一個執(zhí)行Timer次數(shù)的參數(shù)。這個執(zhí)行次數(shù)會以數(shù)組的形式返回每次執(zhí)行的時間 like this

''' 學習中遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：725638078 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ ''' import timeitfoooo = """ sum = [] for i in range(1000):sum.append(i) """print (timeit.repeat(stmt="[i for i in range(1000)]", repeat=2, number=100000))

輸出

[4.466734170913696, 4.255025148391724]

我們可以根據(jù)此，對所有執(zhí)行時間取min最小值，平均值，最大值得到我們想要的數(shù)據(jù)。感覺還是十分方便的。

貼一個稍微進階的例子，就是用到了一下setup參數(shù)的例子，是我在測試protocol buffer python的時候的例子：

# coding: utf-8 import timeit# 初始化類 x = """ say_hi.ParseFromString(p) """y = """ simplejson.loads(x) """print (timeit.timeit(stmt=x, setup="import say_hi_pb2;""say_hi = say_hi_pb2.SayHi();""say_hi.id = 13423;""say_hi.something = 'axiba';""say_hi.extra_info = 'xiba';""p =say_hi.SerializeToString()", number=1000000))print (timeit.timeit(stmt=y, setup="import simplejson; ""json={""'id': 13423,""'something': 'axiba',""'extra_info': 'xiba',""};""x = simplejson.dumps(json)", number=1000000)

另外需要補充一點是，如果你想直接 stmt 那里執(zhí)行函數(shù)。可以把函數(shù)申明在當前文件中，然后在 stmt = ‘func()’ 執(zhí)行函數(shù)。然后使用setup = ‘from __main__ import func’即可，如果要import 多個需要使用setup = from __main__ import func; import simplejson'

總結(jié)

以上是生活随笔為你收集整理的python教程timeit模块的使用教程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。