python - 记录来自 memory_profiler 的报告

Question

我正在使用 memory_profiler 分析我的代码

from memory_profiler import profile

@profile
def whatever():
    ....
    ....

所以，正如你们中的许多人可能知道的那样，我在屏幕上得到了类似这样的输出:

Line #    Mem usage  Increment   Line Contents
==============================================
     3                           @profile
     4      5.97 MB    0.00 MB   def my_func():
     5     13.61 MB    7.64 MB       a = [1] * (10 ** 6)
     6    166.20 MB  152.59 MB       b = [2] * (2 * 10 ** 7)
     7     13.61 MB -152.59 MB       del b
     8     13.61 MB    0.00 MB       return a

我的问题是:

由于@profile 过程需要很多时间，我想知道我是否可以以某种方式记录/存储此输出，并让脚本继续运行，也许是在晚上。

我的想法是在许多 def 函数中使用装饰器 @profile，并以某种方式将所有结果存储在一个 TXT 或许多不同的 TXT 文件中，这并不重要，重要的是这是否可能。

Answer 1

_{^{来自评论:}}

如果你只是运行

run_my_thing > output.txt

在 shell 中，您可以将 stdout 存储在文件中。

这将完全绕过 memory_profiler。显然，仅重定向 stdout 并不理想，但如果用于人工分析，则应该不是什么大问题。