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
– TerryA, Commented Sep 11, 2013 at 9:49

alecxe · Accepted Answer · 2013-09-11 09:50:06Z

87

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, 2013 at 9:50

alecxe

476k127 gold badges1.1k silver badges1.2k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Joe Iddon Over a year ago

Match objects also implement __getitem__, so you can also do x[0].

glglgl · Accepted Answer · 2013-09-11 09:51:25Z

33

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, 2013 at 9:51

glglgl

91.5k13 gold badges157 silver badges230 bronze badges

Comments

Rob Kwasowski · Accepted Answer · 2020-03-13 04:21:18Z

11

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, 2020 at 4:21

Rob Kwasowski

2,7903 gold badges16 silver badges34 bronze badges

answered Feb 21, 2020 at 17:02

user984003

29.9k69 gold badges205 silver badges320 bronze badges

Collectives™ on Stack Overflow

Passing a function to re.sub in Python

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related