python - 如何从脚本运行 IPython 魔术(或为 Python 脚本计时)

Question

IPython %timeit 魔术命令在测量运行某些 Python 代码所需的时间方面做得很好。现在，我想在 Python 脚本中使用类似的东西。我知道 timeit 模块，但是它有几个缺点，例如，如何自适应地选择运行次数？即默认代码

import timeit
t=timeit.Timer("code(f)", "from __main__ import code,f")
t.timeit() 

运行代码百万次。 %timeit IPyhton 魔法命令会自动执行。我建议我可以使用类似 MATLAB 代码的东西 http://www.mathworks.com/matlabcentral/fileexchange/18798

它会自动完成所有工作(并且还会判断该函数的开销是否很大)。

如何从 Python 脚本中调用 %timeit 魔法(或者可能有更好的计时解决方案)？

Answer 1

这在一定程度上取决于您拥有的 IPython 版本。如果你有 1.x:

from IPython import get_ipython
ipython = get_ipython()

如果您有旧版本:

import IPython.core.ipapi  
ipython = IPython.core.ipapi.get()

或

import IPython.ipapi  
ipython = IPython.ipapi.get()

完成后，运行如下魔法命令:

ipython.magic("timeit abs(-42)")

请注意，脚本必须通过 ipython 运行。