python代碼運行效率之timeit模塊

 想要測試一段代碼的運行時間,我們經常是導入時間模塊,然後在代碼的起始處設置一個時間點,結束時設置一個時間點,然後用結束時間減去起始時間,得到的就是代碼運行的時間效率。

但是呢!在python裏面有個很簡單的方法,就是使用timeit模塊,可以測試一段代碼的執行速度,還可以多次測試,不用手動的一遍遍去執行。

下面我就簡單介紹一個timeit模塊中的函數

        class timeit.Timer(stmt='pass', setup='pass', timer=<timer function>)

        Timer是測量小段代碼執行速度的類。

        stmt參數是要測試的代碼語句(statment);

        setup參數是運行代碼時需要的設置;

        timer參數是一個定時器函數,與平臺有關。

timeit.Timer.timeit(number=1000000)

Timer類中測試語句執行速度的對象方法。number參數是測試代碼時的測試次數,默認爲1000000次。方法返回執行代碼的平均耗時,一個float類型的秒數。

from timeit import Timer
#導入timeit模塊
def t1():
    list_1=[]
    for i in range(1000):
        list_1.append(i)
def t2():
    list_2=[]
    for i in range(1000):
        list_2.extend([i])
T1=Timer("t1()","from __main__ import t1")
print("測試1000次的時間是:",T1.timeit(1000))
# T1.timeit(次數)
T2=Timer("t2()","from __main__ import t2")
print("測試1000次的時間是:",T2.timeit(1000))
# T2.timeit(次數)

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章