文字列中で置換したい文字列が複数あるとき、.replace() メソッドを連ねて書くことが多いですが問題もあります。先にreplaceされて出てきた文字列があとのreplaceで対象になってしまう現象ですね。
これがあるので、例えば二つのワードを相互に入れ替えるというのがそう簡単ではありません。
"YES NO YES NO".replace("YES", "NO").replace("NO", "YES")
==> "YES YES YES YES"
# 相互入替になってくれていない！
str.translateというメソッドは同時置換ができるので目標にやや近いのですが、置換対象が文字列でなくて文字なので、2文字以上の文字列を対象にできず、利用シーンは限られます。
カッとなって同時置換してくれるモジュールを作ったので、ご入り用の際はご活用ください。Python2/3両対応で作ってあります。
import re
def multi_replace(string, mapping):
"""
文字列中の複数の文字列やパターンを同時に置換します。
.replace を連ねる場合と違い、ある置換結果がさらに別の置換を受けてしまうことはありません。
よって二つのワードを相互に入れ替えることもできます。
mapping: 検索対象をキー、その置換文字列を値とする辞書。
キーとしてstrまたはreパターンオブジェクトを指定できる。
キーがreパターンオブジェクトの場合は、置換文字列中でグループ参照 '\\1', '\\2',... が有効。
"""
catch_all_pattern = '|'.join(map(to_pattern, mapping))
replacer = make_replacer(mapping)
return re.subn(catch_all_pattern, replacer, string)[0]
_PatternType = type(re.compile('')) # workaround for Python which does't have typing module
def to_pattern(key):
if isinstance(key, str):
return '(' + re.escape(key) + ')'
elif isinstance(key, _PatternType):
return '(' + key.pattern + ')'
else:
raise ValueError(expr(key) + " is not a str, neither re.compile.")
def make_replacer(mapping):
def _replacer(match):
src = match.group(0)
for key, val in mapping.items():
if src == key:
return val
elif isinstance(key, _PatternType) and re.match("(?:" + key.pattern + r")\Z", src): # workaround for Python which doesn't have re.fullmatch
return key.sub(val, src)
return _replacer