Python timeit模組,pythontimeit模組
測試一段代碼的已耗用時間,在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多少遍
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 timeitimport mathimport pprintdef myfun(): for i in range(100): for j in range(2, 10): math.pow(i, 1/j)n = 100t1 = 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這個有意思的模組