来自命令行的Python脚本的时间函数

Question

我想从脚本中计时函数，最好不要修改所述脚本（即添加计时器装饰器或函数）。

Answer 1

从命令行，将timeit模块作为传递附加设置参数 (-s)的脚本运行。设置应该从脚本中导入您想要计时的函数。

最后，记住添加对该函数的调用。

警告：安装程序将运行一次，导入您的脚本。导入运行所有“不受保护的代码”，因此请记住使用if __name__ == "__main__"约定将函数/对象声明与实际执行分开。

想象一下以下 script.py

import time

print('This will be run on import')

def fun():

    pass

def very_complex_function_that_takes_an_hour_to_run():

    time.sleep(3600)

if __name__ == '__main__':

    print("Nothing beyond the if condition will be run on import")

    very_complex_function_that_takes_an_hour_to_run()

让我们计时fun，运行 100 次。

$ python -m timeit -s 'from test import fun' -n 100 'fun()'

输出

This will be run on import

100 loops, best of 5: 66.6 nsec per loop