timeit模块可以用来测试小段代码的运行时间.
使用timeit模块可以有两种方法
直接调用函数
import timeitdef func(): count = 0 for i in range(1000): count += 1print(timeit.timeit(stmt='func()', setup='from __main__ import func', number=100))
生成对象后,再调用对象的timeit方法和repeat方法
import timeitdef func(): count = 0 for i in range(1000): count += 1timeit1 = timeit.Timer(stmt='func()', setup='from __main__ import func') # 默认为1000000次print(timeit1.timeit())