如何限制pycallgraph中的树深度

Question

我正在使用pycallgraph获取脚本中函数的层次结构。但是，结果太深了，无法用于所有模块和子模块。

例如：

import matplotlib.pyplot as plt

def PLot(x,y):
    a=plt.scatter(x,y)
    return a

array1=[2,3,4]
array2=[5,2,3]


from pycallgraph import PyCallGraph
from pycallgraph.output import GraphvizOutput

def main():
    graphviz = GraphvizOutput()
    graphviz.output_file = 'tree.png'

    with PyCallGraph(output=graphviz):
        a=PLot(x=array1,y=array2)
if __name__ == '__main__':
    main()

但是，这会产生非常大的图像，其中包含使用plt.scatter的所有子模块。但是，我只希望获得脚本中的主要功能（在本例中为PLot（））。在图像中，它们以紫色/粉红色显示。