一个文件夹里都是图片,将属于同一个人的图片放到同一个文件夹
import os import sys import shutil """ 将filePath文件下的图片保存在newFilePath文件夹下的相应子文件夹中 pic 是字典,存放每个图片要移到的子文件夹名 """ #将在一个文件夹里的人名图片移动到各自的文件夹里 def moveImg(filePath): files = [] dirs = os.listdir(filePath) for name in dirs: names = filePath + '/' + name if not os.path.isdir(names): files.append(str(name)) for f in files: fl = filePath + '/' + f #图片的绝对路径 fname = str(f).split('_')[0] #图片代表的人名 if fname in dirs: #已经有了人的文件夹 path = filePath + '/' + fname shutil.move(fl,path) # 移动文件或目录 else: #新人 path = filePath + '/' + fname isExists = os.path.exists(path) if not isExists: os.makedirs(path) shutil.move(fl, path) # 移动文件或目录 ''' 每次遍历的对象都是返回的是一个三元组(root,dirs,files) root 所指的是当前正在遍历的这个文件夹的本身的地址 dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录) files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录) ''' # moveImg()的反向操作 def removeImg(filePath): d = [] dirs = os.listdir(filePath) for name in dirs: names = filePath + '/' + name if os.path.isdir(names): d.append(str(name)) for i in range(len(d)): #对于每个人脸文件夹 dirpath = filePath + '/' + str(d[i]) files = os.listdir(dirpath) for f in files: path = dirpath + '/' + str(f) #图片绝对路径 shutil.move(path, filePath) # 移动文件或目录 for name in dirs: names = filePath + '/' + name if os.path.isdir(names): os.removedirs(names) if __name__ == "__main__": #moveImg('/home/double/work/mydarknet/face') removeImg('/home/double/work/mydarknet/face')