Regex zu ersetzen() mit [] um Funktionsargumente

Question

So habe ich diese großen mathematischen Ausdrücke in einer Zeichenfolge gespeichert, die ich aus SymPy habe, und ich versuche es zu formatieren, so dass ich es in Mathematica verwenden kann.

Ich möchte alles, was wie 'sin(arg)' zu Sin[arg] aussieht, ändern, aber ich muss es auch für Cosinus arbeiten lassen. arg kann irgendeine von diesen sein: theta1, theta2, theta3, theta4, theta5.

Es gibt eine Tonne von anderen Klammern in der Zeichenfolge, die ich nicht ersetzen möchte, so dass es nur sin und cos Klammern bewirken muss.

Für S = "cos(theta1)" Ich habe versucht:

S = S.replace("cos", "Cos") 
S = S.replace("sin", "Sin") 
S = re.sub(r"Sin|Cos(\()theta1|theta2|theta3|theta4|theta5", "[", S) 
S = re.sub(r"Sin|Cos\(theta1|theta2|theta3|theta4|theta5(\))", "]", S) 

Und S wird:

'[)' 

Ich dachte, mit den Klammern um die \( und \) bedeuten würde es nur die spezifischen Gruppen ersetzt, aber anscheinend nicht. Gibt es eine andere Funktion als re.sub, die ich verwenden sollte?

P.S. Gibt es eine Möglichkeit, die sin -> Sin Ersatz in die Regex zu quetschen?

Answer 1

def replacer(m): 
    return m.group(1).capitalize()+"["+m.group(2)+"]" 

re.sub("([a-z]+)\(([a-zA-Z0-9 ]*)\)",replacer,"cos(Theta1)") 

Ich denke, ... vielleicht ...

Answer 2

Da Ihr nur Argumente theta1 durch theta5 sein kann, können Sie einfach tun den Ersatz

sin\((theta[1-5])\) 

mit

Sin[\1] 

und

cos\((theta[1-5])\) 

mit

Cos[\1] 

Die \1 ist ein Rückreferenzierung, es Wert aus der ersten angepassten geklammerten Gruppe in der ursprünglichen Zeichenfolge, in diesem Fall Ihr Argument nehmen.

Aber ich würde mit Joran's Antwort gehen.

Answer 3

import re 

if __name__ == '__main__': 
    test = 'sin (theta1)' 
    regex = (
     r'(sin|cos)'  # group # 1: sin or cos 
     r'\s*'   # zero or more spaces 
     r'\('   # opening bracket 
     r'\s*'   # zero or more spaces 
     r'(theta[1-5])' # group #2 with your parameter 
     r'\s*'   # zero or more spaces 
     r'\)'   # closing bracket 
     r'\s*'   # zero or more spaces 
    ) 

    result = re.sub(regex, r'\1[\2]', test,).capitalize() 
    print(result) 

Answer 4

Sie Teile des Musters angeben, die verbraucht wird, nicht (ersetzt) ​​mit Blick behinds (?<=...) und Lookaheads (?=...):

S = "cos(theta1)" 

S = S.replace("cos","Cos").replace("sin","Sin") 

S = re.sub(r"(?<=Sin|Cos)\((?=theta1|theta2|theta3|theta4|theta5)", "[", S) 
S = re.sub(r"(?<=(Sin|Cos)\[(theta1|theta2|theta3|theta4|theta5))\)", "]", S) 

assert S == 'Cos[theta1]'