如何将 %time 的输出发送到 Jupyter/IPython Notebook 中的文件？

Question

魔术命令%time 生成执行给定代码段（语句）到 Jupyter Notebook 中的输出单元所需的时间。我希望将其附加到指定的文件中。

我的目标是找出算法在不同设置下的运行时间。我能够自动化算法的参数，但无法将%time（和%timeit）魔术命令的输出存储在文件中以供进一步处理。

这可能吗？

Answer 1

作为答案转发：

%timeit（但不是%time）有一个-o 选项使它返回一个对象，所以你可以这样做：

res = %timeit -o foo()

然后，您可以从 res 中获取数字并根据需要将它们写入文件。

为了获得更大的灵活性，请使用标准库中的timeit module。