python - 如何在 Python 脚本中将 cProfile.Profile() 的输出保存到 *.prof 文件

Question

您好，我知道如何使用命令行方法来分析 python 脚本，如下所示。

python -m cProfile -o program.prof my_program.py

但是，我正在使用 cProfile 模块对 Python 中的特定代码进行分析，如下所示。

import cProfile, pstats, io
pr = cProfile.Profile()
pr.enable()
# ... do something ...
pr.disable()
s = io.StringIO()
sortby = 'cumulative'
ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
ps.print_stats()
print(s.getvalue())

如何将 cProfile.Profile() 的输出保存到 *.profile 文件而不是使用 pstats .Stats() 分析并打印分析结果。因此我可以使用 SnakeViz 或类似实用程序来直观地分析统计数据。

Answer 1

Profile 类有 a method to write the stats to a file 。您可以使用它将输出保存到文件中。

filename = 'profile.prof'  # You can change this if needed
pr.dump_stats(filename)