自动修改文件夹中的文件名-python

今天让把训练集中的文件名统一一下，于是写了这个改文件夹中文件名的小程序，代码如下

import os

path = input('请输入文件路径(结尾加上/)：')

# 获取该目录下所有文件，存入列表中
fileList = os.listdir(path)

n = 0
for i in fileList:
    # 设置旧文件名（就是路径+文件名）
     # os.sep添加系统分隔符
    oldname = path + os.sep + fileList[n] 
    #.exr 和.hdr是我的训练集文件的扩展名
    h = '.hdr'
    e = '.exr'
    # 设置新文件名
    if e in oldname:
        k = str(n + 1).zfill(5)
        newname = path + os.sep + 'z' + k + '.exr'
    if h in oldname:
        k = str(n + 1).zfill(5)
        newname = path + os.sep + 'z' + k + '.hdr'
    # 用os模块中的rename方法对文件改名
    os.rename(oldname, newname)
    print(oldname, '======>', newname)

    n = n + 1