正则表达式从函数中提取表达式?

时间:2016-03-29 05:01:39

标签: python regex

以下是一些示例输入:

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)]

1 个答案:

答案 0 :(得分:3)

正如评论中所解释的那样,正则表达式不适合这项任务;但是,您可以通过计算括号来轻松解决问题:

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