将在一个文件夹里的人名图片移动到各自的文件夹里及其反操作

一个文件夹里都是图片，将属于同一个人的图片放到同一个文件夹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')