Как ограничить глубину дерева в pycallgraph - PullRequest
Новая
       31

Как ограничить глубину дерева в pycallgraph

0 голосов
/ 26 марта 2019

Я использую 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 ()), которые есть в моем скрипте. На изображении они показаны фиолетовым / розовым цветом.

...