基础用法
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) , '秒')