使用正则表达式切换字符

Question

假设我有类似s='abaabbab' 的东西。是否可以使用正则表达式将其更改为s='babbaaba'？我主要想知道这是否可能，而不是是否合理。

我认为其中一个可能会起作用（之前已导入 re）：

s='ab'
re.sub('a|b',lambda x:['b','a'][x=='a'],s)
#or maybe
re.sub(r'a|b',lambda x:['b','a'][x=='b'],s)

但它们只会产生'bb'。

---

我知道我可以用c 替换所有a，然后用a 替换所有b，然后用b 替换所有c ，但这似乎很长。

gnibbler suggests this:

from string import maketrans
trans_table = maketrans(".-a","-.A")
"foo-bar.".translate(trans_table)

但我似乎无法在 python 3 中工作。

Answer 1

假设我有类似 s='abaabbab' 的东西。是否可以使用正则表达式将其更改为 s='babbaaba' ？我主要想知道这是否可能，而不是是否合理。

是的 - 但您应该使用正确的 str.translate 形式，正如 falsetru 指出的那样...

import re

s = 'abaabbab'
print re.sub('[ab]', lambda L: {'a': 'b', 'b': 'a'}[L.group()], s)
# babbaaba

其工作原理是查找字符（在[] 之间定义的字符），并在匹配时调用替换函数（此处由 lambda 定义），该函数使用字典查找合适的替换字符。

Answer 2

在 Python 3.x 中，您应该传递将 unicode 序数映射到 unicode 字符的字典。 (str.translate documentation)

>>> s = 'abaabbab'
>>> s.translate({ord('a'): 'b', ord('b'): 'a'})
'babbaaba'

为什么 re.sub 代码不起作用？

因为传递给替换函数的参数不是字符串，而是a match object；将匹配对象与字符串进行比较产生False (0)；替换返回第一项 b。

使用group method 获取匹配的字符串。

>>> s = 'ab'
>>> re.sub('[ab]', lambda x: ['b','a'][x.group() != 'a'], s)
'ba'
>>> re.sub('[ab]', lambda x: ['a', 'b'][x.group() == 'a'], s)
'ba'
>>> re.sub('[ab]', lambda x: 'ab'[x.group() == 'a'], s)
'ba'