python批量修改图片名称

1、cnn训练对于训练样本，需要给图片打标签，批量修改图片名称。

上代码：

import os
import re
import sys

def RenameFiles(path):
    fileLists = os.listdir(path)
    #print(fileLists)
    index = 1
    os.chdir(path) 
    for fileName in fileLists:
        print("fileName:", fileName)
        pat = ".+\.(jpg|jpeg|JPG|png)"  # 匹配文件名正则表达式
        pattern = re.findall(pat, fileName)  # 进行匹配
        #print(pattern)
        if (len(pattern)) > 0:
            if index>0 and index < 10:
                os.rename(fileName, ('900'+str(index)+ '.' + pattern[0]))  # 文件重新命名
            elif index>9 and index < 100:
                os.rename(fileName, ('90'+str(index)+ '.' + pattern[0]))  # 文件重新命名
            else :
                os.rename(fileName, ('9'+str(index)+ '.' + pattern[0]))  # 文件重新命名
            index+=1
    print(index)
RenameFiles("/Users/baizhihui/Documents/machine/pkg2/pipeData/train/9")
sys.stdin.flush()  # 刷新