Comment assembler des arbres de syntaxe abstraite Python (AST) dans Mathematica?

Je ne sais pas exactement ce que vous recherchez. Je pense que la réponse à votre question:

Jessaie de générer par programme du code Python en MMA […] puis soumettez-le à Python pour évaluation. Je pourrais assembler des chaînes de code Python, mais ensuite je dois gérer toute lindentation, ce qui semble compliqué.

Est-il possible de créer ce type dobjet Python du côté Mathematica?

est assez simple en utilisant ExternalEvaluate et Python « s  » ast  » bibliothèque.

Voici un exemple:

code = ""[i**2 for i in range(10)]""; astTemplate = StringTemplate["import ast; eval(compile(ast.parse(`1`, mode="eval"), "", "eval"))"]; astTemplate[code] (* "import ast; eval(compile(ast.parse("[i**2 for i in range(10)]", mode="eval"), "", "eval"))" *) ExternalEvaluate["Python", astTemplate[code]] (* {0, 1, 4, 9, 16, 25, 36, 49, 64, 81} *) 

(jai utilisé eval au lieu de exec, car eval renvoie une valeur.)

Commentaires

• Merci Anton. Oui, jaurais pu être plus clair. Permettez-moi de mettre à jour ma question un peu. En gros, je veux générer le  » ‘ [i ** 2 pour i dans lintervalle (10)] ‘  » partie de votre réponse côté MMA, mais pas comme une chaîne Python, mais comme un objet ast.Module qui peut se brancher sur compile et eval et exec.

Je pense que compiler du code Python sous forme de chaîne est en fait plus simple que ce que vous proposez. Mais je sais aussi que le simple fait de dire que cela ne convaincra personne, alors voici un exemple.

Nous définissons quelques têtes symboliques qui représenteront notre programme Python dans Mathematica, et une fonction pour rendre expressions avec ces têtes symboliques:

ToPythonString[statements_List] := StringRiffle[ToPythonString /@ statements, "\n"] ToPythonString[PyBlock[statement_, children_, indent_ : 0]] := StringJoin[ StringRepeat[" ", indent], statement, ":\n", ToPythonString[PyIndent /@ children] ] ToPythonString[PyStatement[statement_, indent_ : 0]] := StringJoin[ StringRepeat[" ", indent], statement ] PyIndent[PyBlock[statement_, children_, indent_ : 0]] := PyBlock[ statement, PyIndent /@ children, indent + 1 ] PyIndent[PyStatement[statement_, indent_ : 0]] := PyStatement[ statement, indent + 1 ] 

Ce que ces fonctions nous permettent de faire est décrire du code Python dans Mathematica sans penser à lindentation, cest un peu comme construction de code Python avec le module ast.

Voici un exemple de rendu de lexpression symbolique sous forme de chaîne:

prog = { PyStatement["a = 1"], PyStatement["b = 2"], PyBlock["If a > b", { PyStatement["Print("a is larger than b")"] }], PyBlock["def f(x)", { PyStatement["Print("executing f")"], PyBlock["if x > 0", { PyStatement["Print("x is larger than 0")"] }] }] }; ToPythonString[prog] 

Out:

a = 1 b = 2 If a > b: Print("a is larger than b") def f(x): Print("executing f") if x > 0: Print("x is larger than 0") 

Nous pouvons facilement construire là-dessus et rendre notre représentation symbolique du programme Python plus descriptive.

PyAssign[lhs_, rhs_] := PyStatement[lhs <> " = " <> rhs] PyPrint[text_] := PyStatement["Print(" <> text <> ")"] PyFunction[name_, args_, statements_] := PyBlock[ "def " <> name <> "(" <> StringRiffle[args, ", "] <> ")", statements ] PyIf[cond_, statements_] := PyBlock[ "If " <> cond, statements ] PyIf[cond_, statements_, elseStatements_] := { PyBlock[ "If " <> cond, statements ], PyBlock[ "else", elseStatements ] } 

Avec ces définitions daide, nous pouvons maintenant écrire le programme suivant dans un style très lisible.

prog = { PyAssign["a", "1"], PyAssign["b", "2"], PyIf[ "a > b", { PyPrint["a is larger than b"] }], PyFunction["f", {"x"}, PyIf[ "x > 0", {PyPrint["x is larger than 0"]}, {PyPrint["x is not larger than 0"]} ] ] }; ToPythonString[prog] 

Out:

a = 1 b = 2 If a > b: Print(a is larger than b) def f(x): If x > 0: Print(x is larger than 0) else: Print(x is not larger than 0) 

Si vous ne lavez pas encore fait, recherchez  » C symbolique dans la documentation Mathematica. Cest essentiellement un moyen de construire un AST pour un programme en langage C dans Mathematica qui peut ensuite être converti en code C exécutable. Cest essentiellement là que nous nous dirigeons avec ce code, bien que si javais lintention de faire une implémentation complète comme celle-ci, cela ne ressemblerait pas exactement à ceci (le module ast vaut certainement la peine dêtre étudié si lon veut suivre la voie). / p>

Revenons au point: ce que je veux transmettre avec cette réponse, cest que vous navez pas à passer beaucoup de temps pour construire un petit framework qui résout plus ou moins le problème dindentation que vous évoquez dans votre question .

Commentaires

• Oui CE cela ressemble beaucoup à ce que jimaginais serait une torture à mettre en œuvre, et vous avez raison, vous lavez fait assez simple. Je vais devoir continuer de cette manière en étendant le langage vers le bas vers les feuilles pour couvrir les atomes et les opérateurs, les listes et les tuples et autres, mais ce cadre gère toute lindentation, et cela devrait être faisable.
• @berniethejet Dans cet exemple, jai défini par exemple PyIf quelque chose qui sévalue en autre chose. Vous pouvez envisager dajouter à la place la définition à ToPythonString, cest-à-dire ToPythonString[PyIf[...]] :=. De cette façon, vous pouvez inspecter lensemble de votre programme sous forme symbolique, sans lavoir évalué jusquà ce que vous appeliez ToPythonString.
• Merci, cest bon à savoir. Votre framework est certainement meilleur que tout ce que jaurais bricolé (grognant tout le long, comme je le serais inévitablement, ennuyé par la croisade dindentation de Guido ‘).