Python lambda with regex

When using re.sub() part of re for python, a function can be used for sub if I am not mistaken. To my knowledge it passes in the match to whatever function is passed for example:

r = re.compile(r'([A-Za-z]')

Is there a smarter way to have it pass in a second arg other than with a lambda that calls a method?

r.sub(lambda x: function(x,arg),string)

Best answer

You can use functools.partial:

>>> from functools import partial
>>> def foo(x, y):
...     print x+y
>>> partial(foo, y=3)
<functools.partial object at 0xb7209f54>
>>> f = partial(foo, y=3)
>>> f(2)

In your example:

def function(x, y):
     pass # ...
r.sub(functools.partial(function, y=arg),string)

And a real use with regexes:

>>> s = "the quick brown fox jumps over the lazy dog"
>>> def capitalize_long(match, length):
...     word =
...     return word.capitalize() if len(word) > length else word
>>> r = re.compile('\w+')
>>> r.sub(partial(capitalize_long, length=3), s)
'the Quick Brown fox Jumps Over the Lazy dog'