Выполнить через AST выражения в Sympy - PullRequest
Новая
       5

Выполнить через AST выражения в Sympy

0 голосов
/ 20 января 2020

Я использую sympy для вычисления некоторых производных высшего порядка выражения сложной функции. Я хотел бы пройти AST выражения, например, go сначала через глубину узлов. Как мне это сделать?

1 Ответ

2 голосов
/ 20 января 2020

Простое перемещение в глубину будет выглядеть так: 

from sympy import pi, sin
from sympy.abc import a,x,y

def depth_first_traverse(expr):
    for arg in expr.args:
        depth_first_traverse(arg, depth+1, new_marks+'+---', parent_ind=ind)
    if len(expr.args) == 0:
        # we reached a leaf of the tree
        pass  # do something with leaf expr
    else:
        pass  # do something with compound expr
depth_first_traverse(sin(a*x*pi+1.5)/y)

С помощью дополнительных параметров можно достичь более сложных целей, например, распечатать дерево в порядке глубины, показывая, как все сходится : 

from sympy import srepr, pi, sin
from sympy.abc import a,x,y

def depth_first_traverse(expr, depth=0, marks='', parent_ind=None):
    for ind, arg in enumerate(expr.args):
        new_marks = marks.replace('+', '|').replace('-', ' ')
        if parent_ind == 0:
            new_marks = new_marks[:-4] + ' ' + new_marks[-3:]
        depth_first_traverse(arg, depth+1, new_marks+'+---', parent_ind=ind)
    if len(expr.args) == 0:
        print(marks, end="> ")
        print("symbol", srepr(expr))
    else:
        print(marks, end="+ ")
        print("function", expr.func, "had", len(expr.args), "arguments")

    print(marks.replace('+', '|').replace('-', ' '))

depth_first_traverse(sin(a*x*pi+1.5)/y)

Вывод: 

    +---> symbol Symbol('y')
    |   
    +---> symbol Integer(-1)
    |   
+---+ function <class 'sympy.core.power.Pow'> had 2 arguments
|   
|       +---> symbol Float('1.5', precision=53)
|       |   
|       |   +---> symbol pi
|       |   |   
|       |   +---> symbol Symbol('a')
|       |   |   
|       |   +---> symbol Symbol('x')
|       |   |   
|       +---+ function <class 'sympy.core.mul.Mul'> had 3 arguments
|       |   
|   +---+ function <class 'sympy.core.add.Add'> had 2 arguments
|   |   
+---+ function sin had 1 arguments
|   
+ function <class 'sympy.core.mul.Mul'> had 2 arguments
...