正则替换

今天遇到一个问题，要求把'A1;A2;A3;B1;B2'这样的一个字符串转换成'A3;A4;A5;B1;B2' ，也就是把所有A字符串后面的数字向上偏移2，如果A字符串比较少并且后面的数字固定的话，用string.replace一个个替换当然是可以的，但是如果不是那样的话就麻烦大了。首先想到的当然是用正则了。看了re模块的sub函数，定义如下：

sub(	pattern, repl, string[, count])

 

从定义上看，不知道如何根据pattern匹配的情况来给定string。google了一下找到了别人的例子，原来sub是把匹配到得match对象传递给了string，string其实可以是个函数，它应该接收一个match对象参数，所以上述问题的解决方法如下：

 

import re

old = 'A1;A2;A3;B1;B2'   
new = re.sub('(A)(/d+)',  lambda m:m.group(1)+str(int(m.group(2))+2),  old)
print new

由于本例子里面对string的处理规则比较简单，所以上面是用了lambda函数；如果替换规则比较复杂的话，则可以定义一个单独的函数处理，如下：

 

import re

def repl(m):

    return m.group(1)+str(int(m.group(2))+2)

old = 'A1;A2;A3;B1;B2'   
new = re.sub('(A)(/d+)',  repl,  old)
print new