关于python timeit的使用

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)

这种情况下只能将需要测试的功能重新定义一个无参数的函数来进行测试。