Python timeit模塊


測試一段代碼的運行時間,在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 timeit
import math
import pprint

def myfun():
    for i in range(100):
        for j in range(2, 10):
            math.pow(i, 1/j)

n = 100

t1 = 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)

D:\005-design\001-python\python.exe C:/Users/Administrator/PycharmProjects/untitled/test.py
0.07670407246477397
[0.05263928378686393,
 0.0564007225542589,
 0.04596403840020144,
 0.08785868398187624,
 0.0807014070693659]
()
0.04607743527370933
[0.045046244306927,
 0.04444133456363253,
 0.044189130814860755,
 0.05559367055818942,
 0.04449898113478035]

Process finished with exit code 0

為了輸出美觀,使用了pprint這個有意思的模塊


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM