仅分析用户定义的函数pstats

时间:2016-04-14 22:33:28

标签: python profiling pstats

我试图从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时,我也会调用库函数。有没有办法可以过滤这些来打印我的函数的调用?

1 个答案:

答案 0 :(得分:2)

您可以通过传入print_stats()函数中的模块名称来过滤模块。例如,假设您的Python文件的名称是" my_python.py",您可以执行print_stats(" my_python")。您还可以传入函数名称。