1、测试单行代码
import timeit
timeit.timeit('x=1',number=10)
2、函数测试
import timeit
def myfunc():
return 100
timeit.timeit('myfunc()','from __main__ import myfunc',number=10)
3、多次测试
import timeit
t = timeit.repeat('myfunc()','from __main__ import myfunc',number=10,repeat=5)
NOTE:
貌似不能传递参数,比如
import timeit
import pandas as pd
df = pd.DataFrame({'a':[1,2,3]})
def myfunc(df):
return 100
timeit.timeit('myfunc(df)','from __main__ import myfunc',number=10)
这种情况下只能将需要测试的功能重新定义一个无参数的函数来进行测试。