timeit模块

timeit模块可以用来测试小段代码的运行时间.

使用timeit模块可以有两种方法

直接调用函数

import timeit


def func():
    count = 0
    for i in range(1000):
        count += 1


print(timeit.timeit(stmt='func()', setup='from __main__ import func', number=100))     

生成对象后,再调用对象的timeit方法和repeat方法

import timeit


def func():
    count = 0
    for i in range(1000):
        count += 1


timeit1 = timeit.Timer(stmt='func()', setup='from __main__ import func') # 默认为1000000次
print(timeit1.timeit())