今天遇到一个问题,要求把'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