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

Question

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

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

这可能吗？

Answer 1

转发作为答案：

%timeit(but not %time) 有一个-o选项使它返回一个对象，所以你可以这样做：

res = %timeit -o foo()

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

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