像python中的正则表达式一样替换ACD | BCD

Question

A，B，C和D代表正则表达式的不同部分。

我要实现的效果：输入字符串是ACD或BCD。 用E替换C后，输出应为AED或BED。

我使用的正则表达式：

r=(A)C(D)|(B)C(D)

但是，当我进行替换时出现了问题。 如果我使用r.sub(r'\\1s0\\2',inputstring)则当输入为BCD时将出现不匹配组错误。 如果我使用r.sub(r'\\3s0\\4',inputstring)则当输入为ACD时会出现组不匹配错误。

那么，如何编辑正则表达式以避免这种情况？

Answer 1

使用(A|B)C(D)代替(A)C(D)|(B)C(D) ：

import re
r = re.compile(r'(A|B)C(D)')
r.sub(r'\1E\2', 'ACD')    # 'AED'
r.sub(r'\1E\2', 'BCD')    # 'BED'

Answer 2

您可以使用替代函数代替字符串。 该函数的返回值用作替换字符串。

import re

def repl(m):
    # m: the matched object.
    if m.group(1) is not None:
        prefix, suffix = m.group(1), m.group(2)
    else:
        prefix, suffix = m.group(3), m.group(4)
    return '{}E{}'.format(prefix, suffix)

re.sub('(A)C(D)|(B)C(D)', repl, 'ACD') # AED
re.sub('(A)C(D)|(B)C(D)', repl, 'BCD') # BED

另外，如果您使用regex模块而不是Python内置的re模块，则可以执行以下操作：

>>> import regex # NOTE: not `re`, but `regex`
>>>
>>> regex.sub('(A)C(D)|(B)C(D)', r'\1\3E\2\4', 'ACD')
'AED'
>>> regex.sub('(A)C(D)|(B)C(D)', r'\1\3E\2\4', 'BCD')
'BED'