Python修改文件夹中图片名并对其进行移动

最近搞泰迪杯，数据增强的时候文件生成目录出了些问题，又不想重新增强，所以写个demo解决一下我的问题。

先看一下我的文件格式，这样的文件夹有40个，每个下边都储存了增强后的image和mask，而且名字乱乱的。


例如文件夹/home/csj/task/task1/output/8中都是如下的形式：
·········
groundtruth(1)_8_8.jpg_000772b0-5fd4-4d11-b1a1-4d7010805447.jpg
8_original_8.jpg_000772b0-5fd4-4d11-b1a1-4d7010805447.jpg
·········
我想将/home/csj/task/task1/output下原始image与掩模图mask分开重命名再储存到/home/csj/task/task1文件夹中的train和label中。
上代码，先将mask的后缀改为png,然后分割字符串匹配改名字移动到对应目录下。

import os
path = r"/home/csj/task/task1/output"
for pa in os.listdir(path):
    pathdir = "/home/csj/task/task1/output/"+pa
    for file in os.listdir(pathdir):
        ext = os.path.splitext(file)
        #print(str(ext).split('_')[-1])
        #print(str(ext).split('_')[1])  #groundtruth or original
        #print(str(ext[0]).split('_')[-1])
        if str(ext).split('_')[1] == 'groundtruth':
            newext = '.png'
            filename = 'label' + str(ext[0]).split('_')[-1] + '.png'
            oldfile = os.path.join(pathdir,file)
            newfile = os.path.join(r"/home/csj/task/task1/label",filename)
            os.rename(oldfile,newfile)
    for file in os.listdir(pathdir):
        ext = os.path.splitext(file)
        if ext[1] == '.jpg':    
            filename = 'train' + str(ext[0]).split('_')[-1] + '.jpg'
            oldfile = os.path.join(pathdir,file)
            newfile = os.path.join(r"/home/csj/task/task1/train",filename)
            os.rename(oldfile,newfile)

最后结果

这样标签和原图就都是一一对应的储存在两个文件夹下了。