Eu precisava de uma solução em que as seqüências a serem substituídas pudessem ser expressões regulares, por exemplo, para ajudar a normalizar um texto longo, substituindo vários caracteres de espaço em branco por um único. Com base em uma cadeia de respostas de outras pessoas, incluindo MiniQuark e mmj, é isso que eu criei:
def multiple_replace(string, reps, re_flags = 0):
""" Transforms string, replacing keys from re_str_dict with values.
reps: dictionary, or list of key-value pairs (to enforce ordering;
earlier items have higher priority).
Keys are used as regular expressions.
re_flags: interpretation of regular expressions, such as re.DOTALL
"""
if isinstance(reps, dict):
reps = reps.items()
pattern = re.compile("|".join("(?P<_%d>%s)" % (i, re_str[0])
for i, re_str in enumerate(reps)),
re_flags)
return pattern.sub(lambda x: reps[int(x.lastgroup[1:])][1], string)
Ele funciona para os exemplos dados em outras respostas, por exemplo:
>>> multiple_replace("(condition1) and --condition2--",
... {"condition1": "", "condition2": "text"})
'() and --text--'
>>> multiple_replace('hello, world', {'hello' : 'goodbye', 'world' : 'earth'})
'goodbye, earth'
>>> multiple_replace("Do you like cafe? No, I prefer tea.",
... {'cafe': 'tea', 'tea': 'cafe', 'like': 'prefer'})
'Do you prefer tea? No, I prefer cafe.'
O principal para mim é que você também pode usar expressões regulares, por exemplo, para substituir apenas palavras inteiras ou para normalizar o espaço em branco:
>>> s = "I don't want to change this name:\n Philip II of Spain"
>>> re_str_dict = {r'\bI\b': 'You', r'[\n\t ]+': ' '}
>>> multiple_replace(s, re_str_dict)
"You don't want to change this name: Philip II of Spain"
Se você quiser usar as teclas de dicionário como seqüências normais, poderá escapar delas antes de chamar multiple_replace usando, por exemplo, esta função:
def escape_keys(d):
""" transform dictionary d by applying re.escape to the keys """
return dict((re.escape(k), v) for k, v in d.items())
>>> multiple_replace(s, escape_keys(re_str_dict))
"I don't want to change this name:\n Philip II of Spain"
A função a seguir pode ajudar a encontrar expressões regulares incorretas entre as chaves do seu dicionário (já que a mensagem de erro de multiple_replace não é muito reveladora):
def check_re_list(re_list):
""" Checks if each regular expression in list is well-formed. """
for i, e in enumerate(re_list):
try:
re.compile(e)
except (TypeError, re.error):
print("Invalid regular expression string "
"at position {}: '{}'".format(i, e))
>>> check_re_list(re_str_dict.keys())
Observe que ele não encadeia as substituições, mas as executa simultaneamente. Isso o torna mais eficiente sem restringir o que ele pode fazer. Para imitar o efeito do encadeamento, talvez seja necessário adicionar mais pares de substituição de cadeia e garantir a ordem esperada dos pares:
>>> multiple_replace("button", {"but": "mut", "mutton": "lamb"})
'mutton'
>>> multiple_replace("button", [("button", "lamb"),
... ("but", "mut"), ("mutton", "lamb")])
'lamb'