如何在Python中的pandas.DataFrame列上的函数中测量经过的时间？

如何解决如何在Python中的pandas.DataFrame列上的函数中测量经过的时间？

我试图找出执行函数以计算pd.DataFrame列上的信息所花费的时间

以下是数据示例：

import pandas as pd
import re
import timeit 

df = pd.DataFrame(['year is 2009','2010'
                    'edition 2008','2010 found'],dtype="string",columns=['message'])
                    
def find_year(x):
    year = re.search('\d{4}',x)                    
    return int(year.group(0)) if year else np.nan

我该如何测量需要多长时间？

df['year'] = df['message'].map(find_year)

在实际代码上，该功能更难，我正在尝试使其尽可能高效。因此，我希望测量执行时间以优化随后应用的功能。

我尝试过：

print("The time taken is ",timeit.timeit(stmt="df['message'].apply(find_year)",number=100))

获得结果：

from timeit import default_timer as timer
start = timer()
df['phase_result'] = df['message'].apply(find_year)
end = timer()
print("The time taken is ",end - start) 

最后一种使用default_timer的方法不允许我包含number=10000使用的参数timeit。