仅分析用户定义的函数 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")。

你也可以传入函数名。