Python性能分析-timeit模塊

基礎用法

import timeit

# 測試的函數
def test1():
    li = [i for i in range(1000)]  # 列表推導式

"""
stmt: 需要測試的函數或語句
setup: 運行的環境
number: 被測試的函數或語句, 執行次數
"""

# 執行test1(), 執行1000次, 所消耗的時間
t1 = timeit.timeit(stmt="test1()", setup="from  __main__ import test1", number=1000)
print('測試test1的時間: ', t1 , '秒')

若需要把這個執行任務, 測試很多遍, timeit中：repeat函數

import timeit

# 測試的函數
def test1():
    li = [i for i in range(1000)]  # 列表推導式

"""
stmt: 需要測試的函數或語句
setup: 運行的環境
number: 被測試的函數或語句, 執行次數
repeat: 測試的次數
"""

# 測試100次->執行test1(), 執行1000次, 所消耗的時間
t1 = timeit.repeat(stmt="test1()", setup="from  __main__ import test1", number=1000, repeat=100)
print('測試100次test1的平均時間: ', sum(t1) / len(t1) , '秒')