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) , '秒')