python re.sub-替代替换模式

Question

我想为re.sub提供替代的替换模式。

假设我有两种搜索模式可供选择，如下所示：

re.sub(r"[A-Z]+|[a-z]+", replacementpattern, string)

而不是提供一个替换模式，我想以某种方式捕获匹配哪个搜索模式的替换项并提供替换模式。这可能吗？谢谢。

PS。这里的代码细节无关紧要，这是一个普遍的问题。

Answer 1

您可以将函数传递给re.sub()。在该函数中，您可以根据捕获的组返回所需的值。一个简单的代码说明：

>>> def fun(m):
...   if m:
...     if m.group(1):
...        return 'x'
...     else:
...        return 'y'


>>>print re.sub(r"([A-Z]+)|([a-z]+)", fun , "ab")

该函数fun()检查匹配是否成功，并根据捕获的组返回替换字符串。如果[A-Z]+匹配，x则为替换字符串，否则[a-z]+为匹配字符串，并且y为替换字符串。

有关更多信息：doc