by HPC_ZY
读取大文件夹里面(包含子文件夹)所有指定类型文件。由于前人数据存放没有规律,非常乱。为了整理到一起,所以需要先获取所有目标类型文件的路径及文件名。作为Python新手,在这里记录分享。
代码
利用os.walk()
找到所有文件名,再利用string.endswith()
判断文件类型,就能提取出来了
import os
# directory-主路径
# fileType-指定文件类型
# fileList-目标类型文件列表(路径+文件名)
def SearchFiles(directory, fileType):
fileList=[]
for root, subDirs, files in os.walk(directory):
for fileName in files:
if fileName.endswith(fileType):
fileList.append(os.path.join(root,fileName))
return fileList
使用
directory = '.\\test'
fileType '.jpg'
fileList = SearchFiles(directory, fileType)
print(fileList)