批量修改文件夹中的文件名

最新推荐文章于 2025-07-02 11:59:30 发布

原创最新推荐文章于 2025-07-02 11:59:30 发布 · 290 阅读

2 ·

CC 4.0 BY-SA版权

DataSet 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一个使用Python编写的脚本，该脚本能够批量地按照指定前缀和模式重命名目录下的文件，适用于数据集预处理场景。通过命令行参数输入路径、文件格式和前缀，可以轻松实现文件名的统一修改。

前言：

一切为了数据集准备。。。

代码：

import os
import re
import sys
import argparse

parser = argparse.ArgumentParser(description="change img&xml name")
parser.add_argument("--input_path",type=str,help="The path of the input img/xml")
parser.add_argument("--pattern",type=str,help="The format of the file")
parser.add_argument("--add_name_prefix",type=str,help="The name of the prefix of the file")
args = parser.parse_args()

def rename_all(path,pattern,prefix):
    file_list = os.listdir(path)
    print('*' * 100)
    print("before rename:" + str(file_list))
    current_path = os.getcwd()
    os.chdir(path)
    num = 1
    for file_name in file_list:
        pat = "."+str(pattern)
        pattern_result = re.findall(pat, file_name)
        os.rename(file_name, (str(prefix) + str(num) + pattern_result[0]))
        num = num + 1
    os.chdir(current_path)
    sys.stdin.flush()
    print('*' * 100)
    print("after rename:" + str(os.listdir(path)))

if __name__ == "__main__":
    rename_all(args.input_path,args.pattern,args.add_name_prefix)