Passing a function to re.sub in Python

Question

I have strings that contain a number somewhere in them and I'm trying to replace this number with their word notation (ie. 3 -> three). I have a function that does this. The problem now is finding the number inside the string, while keeping the rest of the string intact. For this, I opted to use the re.sub function, which can accept a "callable". However, the object passed to it is the internal _sre.SRE_Match and I'm not sure how to handle it. My function accepts a number or its string representation.

How should I write some helper function which can be used to bridge the re.sub call with my function doing the required processing? Alternatively, is there a better way to do what I want?

With a `_sre.SRE_Match`, you can call `.group()` on it to get the groupped items. Is this what you need? — TerryA, Sep 11 '13 at 09:49

score 80 · Accepted Answer · answered Sep 11 '13 at 09:50

80

You should call group() to get the matching string:

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)

prints:

one testing two three

answered Sep 11 '13 at 09:50

alecxe

462,703
120
1,088
1,195

1

Match objects also implement `__getitem__`, so you can also do `x[0]`. – Joe Iddon Sep 04 '18 at 14:43

score 30 · Answer 2 · answered Sep 11 '13 at 09:51

30

To make your function fit with re.sub, you can wrap it with a lambda:

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

answered Sep 11 '13 at 09:51

glglgl

89,107
13
149
217

score 7 · Answer 3 · edited Mar 13 '20 at 04:21

7

A solution without 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)

edited Mar 13 '20 at 04:21

Rob Kwasowski

2,690
3
13
32

answered Feb 21 '20 at 17:02

user984003

28,050
64
189
285

Passing a function to re.sub in Python

3 Answers3

Linked

Related