在 Python 中将函数传递给 re.sub答案

【问题标题】：Passing a function to re.sub in Python在 Python 中将函数传递给 re.sub
【发布时间】：2013-09-15 06:45:09
【问题描述】：

我的字符串中某处包含一个数字，我正在尝试用它们的文字符号替换这个数字（即 3 -> 三）。我有一个功能可以做到这一点。现在的问题是在字符串中找到数字，同时保持字符串的其余部分完好无损。为此，我选择使用re.sub 函数，它可以接受“可调用”。但是，传递给它的对象是内部的_sre.SRE_Match，我不确定如何处理它。我的函数接受一个数字或其字符串表示。

我应该如何编写一些辅助函数来桥接re.sub 调用与我的函数进行所需的处理？或者，有没有更好的方法来做我想做的事？

【问题讨论】：

使用_sre.SRE_Match，您可以在其上调用.group() 以获取分组项目。这是你需要的吗？

标签： python regex

【解决方案1】：

您应该调用group() 来获取匹配的字符串：

import re

number_mapping = {'1': 'one',
                  '2': 'two',
                  '3': 'three'}
s = "1 testing 2 3"

print re.sub(r'\d', lambda x: number_mapping[x.group()], s)

打印：

one testing two three

【讨论】：

Match 对象也实现了__getitem__，所以你也可以实现x[0]。

【解决方案2】：

要使您的函数适合 re.sub，您可以用 lambda 包装它：

re.sub('pattern', lambda m: myfunction(m.group()), 'text')

【讨论】：

【解决方案3】：

没有 lambda 的解决方案

import re

def convert_func(matchobj):
    m =  matchobj.group(0)
    map = {'7': 'seven',
           '8': 'eight',
           '9': 'nine'}
    return map[m]

line = "7 ate 9"
new_line =  re.sub("[7-9]", convert_func, line)

【讨论】：