Pythonでの正規表現を用いた置換について

Question

Pythonで正規表現を用いて置換する場合、調べた限りre.sub関数を用いるのが一般的なようですが、これが関数なのでメソッドチェーン使えないのが不便に感じています。
Javascriptだと

str
.replace(/^あいうえお$/, 'aiueo')
.reaplace(/^かきくけこ$/, 'kakikukeko');

のようにメソッドチェーンが使用できてコードを少なくキレイにかけるのですが、Pythonの場合

str = re.sub('^あいうえお$', 'aiueo', str)
str = re.sub('^かきくけこ$', 'kakikukeko', str)

のように冗長なコードになってしまいます。

何かキレイに記述するご助言があればいただきたいです。

Comments

[otn]

回答では無いですが、かつ、ご存じな気もしますが、メソッドチェーンでコンパクトにコードを書きたければ、Rubyを使いましょう。

Answer 1

既にクローズしていますが、reduceを使用して順番に置換していく方法もあると思います。

import re
from functools import reduce

s = 'あいうえおかきくけこ'
patterns = [['あいうえお', 'aiueo'], ['かきくけこ', 'kakikukeko']]
t = reduce(lambda txt, p: re.sub(*p, txt), patterns, s)
print(t)  # aiueokakikukeko

Answer 2

今後、変換対象と変換後の文字列のペアが追加される（たとえば、'さしすせそ' を 'sasisuseso' にする変換を追加する、等）ことがあるのであれば、変換対象の文字列と変換後の文字列の対応を辞書

repls = {
  'あいうえお': 'auieo', 
  'かきくけこ': 'kakikukeko',
  'さしすせそ': 'sasisuseso',  
}

で持っておいて

regexp = '^(%s)$' % '|'.join(repls.keys())
replacer = lambda m: repls[m[1]]

というのを作っておいて

str = re.sub(regexp, replacer, str)

としますかね〜。実行例 👉 https://replit.com/@su507/ps549k9ndm7hgf#main.py

Comments

[chunchunchun]

コールバック関数を使えるの知りませんでした💦
辞書にまとめてしまえば、何をどう置換してるのか直感的に分かって良さそうですね。
今後使っていこうと思います！

Answer 3

str 型にはカスタムメソッドを追加できないで、継承クラスを定義するなど。

import re

class string(str):
    def subst(self, reg, repl):
        return __class__(re.sub(reg, repl, self))

if __name__ == '__main__':
    text = 'あいうえお'
    result = string(text).subst(r'^あいうえお$', 'aiueo').subst(r'^かきくけこ$', 'kakikukeko')
    print(result)    

    text = 'かきくけこ'
    result = string(text).subst(r'^あいうえお$', 'aiueo').subst(r'^かきくけこ$', 'kakikukeko')
    print(result)    

Comments

[chunchunchun]

一度クラスを定義してしまえば色んなとこで使い回せるので良さそうですね。
なによりメソッドチェーンで記述できるのが魅力的です。
使ってみます！