Python 文字列の置換

Question

前提・実現したいこと

python 2.7です。

文字列の置換を綺麗に行いたいです。
正規表現とかでもっときれいになるとは思うんですが思い浮かびません。

該当のソースコード

file_name = "ABC01_de02_v00.txt"
file_name, ext = os.path.splitext(file_name)
split_name = file_name.split("_")
file_name = ""
for i in range(len(split_name)):
    if i == len(split_name)-1:
        file_name += "fix"
    else:
        file_name += split_name[i] + "_"

最終的にはfile_nameがABC01_de02_endになるようにしたいです。
_v??のところを_endに置換したいです。
??のところはバージョンなので幾つになるかはわかりません。

ただ、ABC01_de02に必ず_v??が含まれることがないとは言い切れません。
例) uv01_wxyz などです。

replace()だと例の際に置換がかかる可能性があり上記のように処理しています。

Answer 1

ただ、ABC01_de02に必ず_v??が含まれることがないとは言い切れません。

例) uv01_wxyz などです。

ちょっとここが理解できなかったのですが、勝手な解釈で_v00自体が含まれない場合もあるという事ですかね？

import os
import re

file_name = "ABC01_de02.txt"
file_name, ext = os.path.splitext(file_name)

check = re.search('(?<=_)v\d{2}', file_name)

if check is None:
    file_name += '_end' + ext
else:
    file_name = re.sub(
        r'(?<=_)v\d{2}', r'end',
        file_name
    ) + ext

print(file_name)

Answer 2

どの程度まで厳密に判定するか？によりますが、とりあえず。

import re
for s in ['ok_v00.txt','ok_v1.txt','ng_v.txt','ng.txt']:
    ret = re.sub(r'_v\d+', '_end', s)
    print(ret)

Comments

[suvera]

ありがとうございます!これだけでもかなりスッキリします。
後方から1つだけしか置換しないということはできますか?
ng_v00ng_v00.txtのような名前にも対応できるようにしたいです。

[can110]

".txt"の手前の"."も含めるように
`ret = re.sub(r'_v\d+.', '_end.', s)`
とすればひとまず対応できます。

[suvera]

なるほど、ありがとうございます。
拡張しない場合は $ 入れればできそうです。
ありがとうございました!

Answer 3

やりたいのはこういうことですかね？

import re

file_name = "ABC01_de02_v00.txt"
file_name2 = re.sub("_v\d\d", "_end", file_name)
print(file_name2)

Answer 4

一応これならABC01_de02_v00.txtもuv01_wxyz_v00.txtも置換できます。

import re

file_name = 'ABC01_de02_v00.txt'
file_name = re.sub(
    r'(?<=_)v\d{2}(?=.txt)', r'end',
    file_name
)

print file_name

Wandbox