python实现将子文件夹下的图片复制到新的文件夹并重命名

一、要求

       在文件夹中有若干子文件夹，且子文件夹中有若干图片，按照要求将子文件夹中符合命名的图片复制到新的文件夹中，并重新命名这些图片。

       即将在train文件夹中有若干命名为rgb_0001_json的子文件中的label.png图片复制到mask文件夹中，并重命名为rgb_0001.png。

import os
import shutil

path = 'c:/Users/Administrator/Desktop/train'
new_path = 'c:/Users/Administrator/Desktop/mask'
count = os.listdir(path)
for j in range(1,len(count)+1):
    for root, dirs, files in os.walk(path):
        if len(dirs) == 0:
            for i in range(len(files)):
                print("i=",i)
                if files[i].find('label.png')!=-1:
                    shutil.copy(os.path.join(path+'/rgb_'+str(j).zfill(4)+'_json/',files[i]), os.path.join(new_path,'rgb_'+str(j).zfill(4)+'.png'))

其中，path是原始文件夹的路径，new_path是新的文件夹路径。

二、shutil.copy

shutil.copy()具体用法：

shutil.copy(source,destination)

source/destination都是字符串形式的路径，其中destination是：

（1）可以是一个文件的名称，则将source文件复制为新名称的destination

（2）可以是一个文件夹，则将source文件复制到destination中

（3）若这个文件夹不存在，则将source目标文件内的内容复制到destination中

三、可能出现的错误

OSError invalid argument

参考：可能是文件路径有问题，python中\是转义字符，路径可以直接写成：C:/Users/Administrator/Desktop/mask。

扫描二维码关注公众号，回复： 2826726 查看本文章