利用os.walk可以不断遍历多层文件夹,直到找到符合要求的文件类型
其中涉及三个参数:
root:是当前循环所在的目录
dirs:是当前目录下的所有文件夹
files:是当前目录下的所有文件
def countImgNum():
count =0
src_path = '/home/ok/imgPath'
#root是当前循环的目录
#dirs是当前目录下的所有文件夹
#files是当前目录下的所有文件
for root,dirs,files in os.walk(src_path,topdown=True):
if not len(dirs) and len(files):
for file in files:
if file[file.rfind('.') + 1:] != 'jpg':
print('regular:%s' % (file[file.rfind('.') + 1:]))
continue
count += 1
srcName = os.path.join(root, file)
print(srcName)
print('all_img_count:%d'%count)