python对文件进行批量重命名

以mat文件为例

原文件名如下图：

现在变成下图：

代码如下所示：

import os

if __name__ == '__main__':
    root = "E:/data"

    path= os.path.join(root, 'UCF-QNRF_pro/train_data/ground_truth')
   
    count = 0
    file_list = os.listdir(path)
    for mat in file_list:
        old_dir = os.path.join(path, mat)
        if os.path.isdir(old_dir):
            continue
        new_dir = os.path.join(path, str(count + 1) + '.mat')
        os.rename(old_dir, new_dir)
        count+=1