如何存储来自%% timeit cell magic的结果?

时间:2016-12-27 19:26:32

标签: python ipython ipython-magic

我无法弄清楚如何存储细胞魔法的结果 - %%timeit?我读过:

  1. Can you capture the output of ipython's magic methods?
  2. Capture the result of an IPython magic function
  3. 并且在这个问题中只回答关于线魔术的问题。在线模式(%)中,这有效:

    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)
    

    它只是执行单元格,没有魔法。这是一个错误还是我做错了什么?

1 个答案:

答案 0 :(得分:10)

您可以在_单元格之后使用%%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