我想要 Python 类/函数来近似 IPython %%timeit

如何解决我想要 Python 类/函数来近似 IPython %%timeit

%%timeit 在 IPython 或 Jupyter notebook 中有两个我喜欢的功能。

1. 运行有限时间，自动调整给定批次数的迭代次数。

2. 输出是平均值 + 标准差。这比多次运行中的最短运行时间要好。

这是来自 IPython 终端的示例输出。第一个来自函数的“慢版本”，第二个来自更快的版本。

In [21]:  %timeit hh = gg.get_map(form="232",year=2019,source="form")
1.82 s ± 111 ms per loop (mean ± std. dev. of 7 runs,1 loop each)

In [18]: %timeit jj = fm.get_map(form="232",source="form")
324 µs ± 22.9 µs per loop (mean ± std. dev. of 7 runs,1000 loops each)

请注意，当函数很慢时，它返回 7 x 1 的执行时间，但是当它很快时，它会拨号到 7 x 1000。

我一直在研究 IPython 代码以了解它们是如何做到的，认为我可以适应一段 Python 代码。这是一项艰巨的工作！与 AST 交互让我进入了一个陌生的领域。

在我看来，以前一定有人这样做过。没有 Python 包吗？

解决方法

简短的回答是，您不能在纯 Python 代码的 py 文件/块中执行此操作。

一旦文件被解析，你就无法从 Python 取回 AST。这就是“宏”是 Python 中经常要求的功能的原因之一，因为它们可以让你做这样的事情。

可能有一些hacky的方法，但粗略地说，您需要找到定义代码的文件，并将内容作为字符串来操作； IPython 在进行自己的解析时可以做什么。

.... 或者你制作了一个 API，它接受一个多行字符串而不是一个 Python 代码块，这很丑但可以工作。

更一般地说，如果您认为我们可以在 IPython 中使代码块更具可重用性，请随时提出问题/拉取请求。