python - 仅分析用户定义的函数 pstats

Question

我正在尝试使用 pstats 模块中的 Stats() ，如下所示:

    p = Profile()
    p.runcall(wrangle_file,input_filename="test.csv",output_file="solution.csv",metrics=True)
    stats = Stats(p)
    stats.strip_dirs()
    stats.sort_stats('cumulative')
    stats.print_stats()

但是，当我执行 print_stats 时，我也会收到对库函数的调用。有没有办法可以过滤这些以仅打印对我的函数的调用？

Answer 1

您可以通过在 print_stats() 函数中传递模块名称来过滤模块。

假设您的 Python 文件的名称是 my_python.py，您需要执行 print_stats("my_python")。

您还可以传入函数名称。