python遍历文件目录，在指定的目录下操作

import  os
"""
这段代码的功能是，在多级目录下，找到目录为（文件夹）Enroll,删除多余的图片文件后只保留一张图片在
Enroll目录下

"""
def searchDir(rootDir):
    for dir_or_file in os.listdir(rootDir):
        filePath = os.path.join(rootDir, dir_or_file) #dir_or_file 的绝对路径
        #print(dir_or_file)

        #判断是否是文件
        if os.path.isfile(filePath): 
            pass   #不作任何操作
            
        #判断是否是文件夹且文件夹名为“Enroll”
        elif os.path.isdir(filePath)and str(dir_or_file)== "Enroll": 
            #print(len(os.listdir(filePath)))
			#在Enroll文件夹下的所有文件，这些文件是一个列表放在all_enrolls
            all_enrolls = os.listdir(filePath)
            #利用切片选取第二个文件及后面的文件
            rm_enrolls = all_enrolls[1:]

			#在Enroll文件夹下只留一个文件，其他文件的删除
            for rm_enroll in rm_enrolls:
                os.remove(os.path.join(filePath, rm_enroll))
                
        #判断是文件夹但文件夹名**不是** “Enroll”  ，不是的话，继续往下遍历目录
        elif os.path.isdir(filePath)and str(dir_or_file)!= "Enroll":
            searchDirFile(filePath)
            #print(filePath)

        else:
            print('not file and dir '+os.path.basename(filePath))





if __name__ == '__main__':
     rootDir=r'C:\XXX\桌面\整合数据集2' #绝对路径
     searchDir(rootDir)