Python -- 同类图像的批量重命名

#coding = utf-8
import os

#给出需要重命名的图像路径data_dir
img_dir = '...'
#定义一个重命名的函数
def rename(data_dir, category): #data_dir--图像的路径，category--图像所属的类
    i = 0
    for file in os.listdir(data_dir): #用于遍历路径中的所有图像
        i = i+1
        #dir--需要更新的图像路径，new_dir--新的图像路径
        dir = os.path.join(os.path.abspath(data_dir), file)
        new_dir = os.path.join(os.path.abspath(data_dir), category+str(i)+'.jpg')
        os.rename(dir, new_dir)
    print('总共有'+str(i)+'张图片')
rename(img_dir, 'aeroplane')    #假定路径下都是飞机的图像