python - 正则表达式从函数中提取表达式？

Question

以下是一些示例输入:

Fxfu(x1,Y,n<X+Y)

Fxfu(x1,Y(n,A,B),Z(n,A,B)+M(n,A,B)<n)

Fxfu(x1,Y(n),Z(n)^M(n)<X+Y)

Fxfu(x1,(n,A,B),Z(n)^M(n)<T(n,A,B))

我希望它们相应的输出是:

n<X+Y

Z(n,A,B)+M(n,A,B)<n

Z(n)^M(n)<X+Y

Z(n)^M(n)<T(n,A,B)

我编写了以下代码

find=regex.compile(r'(Fxfu\(x\d+[,]([Y]\((.*)\)[,]))|Fxfu\(x\d+[,]([Y][,])')
    group = find.search(expression)
    print expression.replace(group.group(0),"")

有什么更好的方法来执行上述操作，以便我可以按如下方式提取数组中的所有参数

[x1,Y,n<X+Y]
[x1,Y(n,A,B),Z(n,A,B)+M(n,A,B)<n]
[x1,Y(n),Z(n)^M(n)<X+Y]
[x1,(n,A,B),Z(n)^M(n)<T(n,A,B)]

Answer 1

正如评论中所解释的，正则表达式不适合这项任务；但是，您可以通过计算括号轻松地一次解决问题:

def extract_args(expr):
    paren = 0
    start = 0
    ret = []
    for i, c in enumerate(expr):
        if c=='(':
            paren+=1
            if paren==1:
                start=i+1
        elif c==')':
            if paren==1 and start:
                ret.append(expr[start: i]) 
            paren-=1
        elif c==',' and paren==1:
            ret.append(expr[start:i])
            start=i+1
    return ret