python正則表達式：將「frame_20」更改爲「frame_0020」的腳本

Question

所以我有一個文件名列表。他們被稱爲：

"fgmask_frame_20.jpg","fgmask_frame_40.jpg",..."fgmask_frame_2000.jpg" etc. 

我想要做的就是讓所有的數字4位數的，所以，如果是frame_20然後使它frame_0020，而frame_0200代替frame_200等

如何我可以在Python中執行此操作嗎？

Answer 1

你可以使用正則表達式如下：

import re 

file_names = ["fgmask_frame_20.jpg", "fgmask_frame_40.jpg", "fgmask_frame_2000.jpg"] 
file_names = [re.sub(r'(.*?_)(\d+)(\.jpg)', lambda x: '{}{:04}{}'.format(x.group(1), int(x.group(2)), x.group(3)), file_name) for file_name in file_names] 
print file_names 

這會爲您提供以下的輸出：

['fgmask_frame_0020.jpg', 'fgmask_frame_0040.jpg', 'fgmask_frame_2000.jpg']