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

Question

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

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

这可能吗？

Answer 1

重新发布作为答案:

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

res = %timeit -o foo()

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

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