如何存储 %%timeit 单元魔法的结果？答案

【问题标题】：How to store the result from %%timeit cell magic?如何存储 %%timeit 单元魔法的结果？
【发布时间】：2016-12-27 19:26:32
【问题描述】：

我不知道如何存储单元魔法的结果 - %%timeit？我读过：

在这个问题中只回答关于线魔法的问题。在线路模式 (%) 下可以：

In[1]: res = %timeit -o np.linalg.inv(A)

但在单元格模式下 (%%) 它没有：

In[2]: res = %%timeit -o 
       A = np.mat('1 2 3; 7 4 9; 5 6 1')
       np.linalg.inv(A)

它只是执行单元格，没有魔法。是错误还是我做错了什么？

【问题讨论】：

我使用的是 IPython 5.1.0
感谢您不仅参考了相关问题，还总结了它们的相关性。您会惊讶（或不惊讶）有多少人在遇到问题时没有这种反应。

标签： python ipython ipython-magic

【解决方案1】：

您可以在%%timeit -o 单元格之后使用_ 变量（存储最后的结果）并将其分配给一些可重用的变量：

In[2]: %%timeit -o 
       A = np.mat('1 2 3; 7 4 9; 5 6 1')
       np.linalg.inv(A)
Out[2]: blabla
        <TimeitResult : 1 loop, best of 3: 588 µs per loop>

In[3]: res = _

In[4]: res
Out[4]: <TimeitResult : 1 loop, best of 3: 588 µs per loop>

我认为这不是错误，因为单元格模式命令必须是该单元格中的第一个命令，因此您不能在该命令前面放置任何内容（甚至 res = ...）。

但是您仍然需要-o，因为否则_ 变量包含None。

【讨论】：

而res 有许多属性，您可以通过res.<tab> 或vars(res) 看到这些属性。

【解决方案2】：

如果你只关心细胞魔法的输出，例如出于记录目的 - 您不需要 TimeitResult 对象中包含的额外元数据，您也可以将其与 %%capture 结合使用：

%%capture result
%%timeit

A = np.mat('1 2 3; 7 4 9; 5 6 1')
np.linalg.inv(A)

然后您可以从 result.stdout 中获取输出，这将产生单元格的任何输出 - 包括计时结果。

print(result.stdout)

'26.4 us +- 329 ns per loop (mean +- std. dev. of 7 runs, 10000 loops each)\n'

这适用于任意单元格魔法，并且可以在下划线解决方案不起作用时作为后备。

【讨论】：