A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

 找回密码
 加入黑马

QQ登录

只需一步,快速开始

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

       即将在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.find('label.png')!=-1:
                    shutil.copy(os.path.join(path+'/rgb_'+str(j).zfill(4)+'_json/',files), 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。
---------------------
作者:蹦跶的小羊羔
来源:CSDN
原文:https://blog.csdn.net/yql_617540298/article/details/81612595
版权声明:本文为博主原创文章,转载请附上博文链接!

2 个回复

正序浏览
回复 使用道具 举报
回复 使用道具 举报
您需要登录后才可以回帖 登录 | 加入黑马