timeit是Python標(biāo)準(zhǔn)庫內(nèi)置的小工具，可以快速測試小段代碼的性能。

認(rèn)識timeit

timeit 函數(shù)：

timeit.timeit(stmt, setup,timer, number)

參數(shù)說明：

• stmt: statement的縮寫，你要測試的代碼或者語句，純文本，默認(rèn)值是 “pass”
• setup: 在運(yùn)行stmt前的配置語句，純文本，默認(rèn)值也是 “pass”
• timer: 計(jì)時(shí)器，一般忽略這個(gè)參數(shù)
• number: stmt執(zhí)行的次數(shù)，默認(rèn)是1000000，一百萬

repeat 函數(shù)：

timeit.repeat(stmt, setup, timer, repeat, number)

是timeit的repeat版，可以指定重復(fù)timeit的次數(shù)，默認(rèn)是3次，然后返回一個(gè)數(shù)組。

舉一個(gè)簡單的例子來說明用法：

import timeit print(timeit.timeit('output = 10*5')) # 0.014560436829924583 print(timeit.repeat('output = 10*5')) # [0.01492984383367002, 0.01342877489514649, 0.013638464966788888]

嗯，看上去沒毛病，實(shí)際上誰也不會(huì)去測沒有意義的加減乘除，我們需要測試自己的代碼。

測試多行代碼

測試多行代碼可以用分號來連接語句。

print(timeit.timeit(stmt='a=10;b=10;sum=a+b'))

也可以用三引號來寫stmt。

import timeit import_module = "import random" testcode = ''' def test(): return random.randint(10, 100) ''' print(timeit.repeat(stmt=testcode, setup=import_module))

但是其實(shí)都挺扯的，自己的代碼會(huì)那么簡單？我們是模塊化編程。

測試模塊中的函數(shù)

如果你要測試的函數(shù)都在一個(gè)模塊里，可以這樣寫timeit。

''' 遇到問題沒人解答？小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群：531509025 尋找有志同道合的小伙伴，互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書！ ''' import timeit import random import arrow# 本地函數(shù) def stupid1():return random.randint(1, 10)# 依賴其他函數(shù) def stupid2():return stupid1()# 依賴其他包或者模塊 def stupid3():return arrow.now()print(timeit.timeit('stupid1()', setup='from __main__ import stupid1')) print(timeit.timeit('stupid2()', setup='from __main__ import stupid2')) print(timeit.timeit('stupid3()', setup='from __main__ import stupid3', number=100))

寫成上面這樣的其實(shí)還是單行的模式。

借用default_timer

timeit自帶的default_timer可以借來用一下。

import timeit import randomdef test(): return random.randint(10, 100)starttime = timeit.default_timer() print("The start time is :",starttime) test() print("The time difference is :", timeit.default_timer() - starttime)

命令行的用法

timeit還支持命令行的調(diào)用方式，不過我覺得太累了，沒必要去嘗試。

C:\pythontest>python -m timeit -s 'text="hello world"' 20000000 loops, best of 5: 13.1 nsec per loop

分享一個(gè)案例

2月29那天，我想今年是閏年啊，計(jì)算閏年有幾種算法啊？孔乙己說有3種：

''' 遇到問題沒人解答？小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群：531509025 尋找有志同道合的小伙伴，互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書！ ''' def is_leap_year_0(year):if year % 4 == 0:if year % 100 == 0:if year % 400 == 0:return Trueelse:return Falseelse:return Trueelse:return Falsedef is_leap_year_1(year):return year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)def is_leap_year_2(year):if year % 400 == 0:return Trueif year % 100 == 0:return Falseif year % 4 == 0:return Truereturn False

這三種方法那種最好啊？這個(gè)不能一概而論吧，因?yàn)橐茨愕膮?shù)是什么，比如1991年不是閏年，方法0和方法1直接就返回了，但方法2還需要走到最后一個(gè)if才知道不是閏年。再比如2020年，方法2直接就返回了，但是方法0和1需要走到最里層的if才得到結(jié)果。所以我們需要取樣測試才公平，比如從1900年到2900年，每個(gè)函數(shù)都跑10000遍。

timeit就不太方便了，它接受的參數(shù)哪能那么復(fù)雜，我們需要包裝一下。

def perf_test(method):years = range(1900, 2900)if method == 0:for y in years:is_leap_year_0(y)if method == 1:for y in years:is_leap_year_1(y)if method == 2:for y in years:is_leap_year_2(y)print(timeit('perf_test(0)', setup='from __main__ import perf_test', number=10000)) print(timeit('perf_test(1)', setup='from __main__ import perf_test', number=10000)) print(timeit('perf_test(2)', setup='from __main__ import perf_test', number=10000))

你們猜猜看哪個(gè)方法結(jié)果最好？你一定想不到。

1.6432780250906944 1.7527272370643914 0.0023122059646993876

總結(jié)

以上是生活随笔為你收集整理的使用timeit测试Python函数的性能的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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