python - 如何从 pycallgraph 中删除库模块或特定函数

Question

我正在使用pycallgraph来分析我的代码性能。但是，调用图非常混乱，有许多对系统函数的调用以及我不想记录的某些函数。如何阻止 pycallgraph 报告这些调用？

Answer 1

Pycallgraph 提供过滤功能来过滤掉您想从调用图中排除的任何模块、类或函数。应在开始跟踪之前定义以下函数并将其传递给 pycallgraph

例子

def filtercalls(call_stack, modul, clas, func, full):
    mod_ignore = ['shutil','scipy.optimize','re','os','sys','json']
    func_ignore = ['CustomFunctionName','pdbcall']
    clas_ignore = ['pdb']
    return modul not in mod_ignore and func not in func_ignore and clas not in clas_ignore

pycallgraph 跟踪开始是

pycallgraph.start_trace(filter_func=filtercalls)

这样，您在 filtercalls 中提供的任何模块、类或函数都将被删除。请注意，在标准库中很多时候只提供模块名称是不够的。因此，在 mod_ignore 中包含 numpy 仍将导致包含 numpy.core