遍历读取文件夹下的所有文件
例如,读取文件夹下,子文件夹的所有的jpg文件:
import glob
path = "./database/20230302/night/*/*.jpg"#设置自己的文件夹路径以及文件
image_files = glob.glob(path, recursive=True)
for image_file in image_files:
image = cv2.imread(image_file)
上面是通过cv2、glob来获取文件夹中所有子文件下的jpg文件,同理,仍可采用此方式来获取其他类型的文件或所有类型的文件
所以:
import glob
#获取指定目录下的所有图片
print glob.glob(r"E:/Picture/*/*.jpg")
#获取上级目录的所有.py文件
print glob.glob(r'../*.py') #相对路径
其中,通过查阅发现还有其他方案列出目录中的文件:
(1)使用os.listdir()
import os
os.listdir()#打印当前路径下所有文件和目录的名称
os.listdir(path)#基于特定路径来打印结果,path是自己设定的
files = [f for f in os.listdir() if os.path.isfile(f)]
#只想打印所有文件不包含目录,那么我们可以使用os.path.isfile() 来进行相应的过滤
#对于目录,同样可以使用函数os.path.isdir() 进行过滤
files = [f for f in os.listdir() if os.path.isdir(f)]
(2)使用os.walk()
调用os.walk() 函数时,它将返回一个生成器。
此时每次调用next() 方法生成下一个值时,它都会进入到一个layer ,结果是一个包含3个项的元组:(dirpath、dirname、filename) 。
举例,如果要获取第3层中所有文件夹的名称,代码如下所示:
from os import walk
f = []
layer = 1
w = walk("/Users/slw")#设置你要查看的路径
for (dirpath, dirnames, filenames) in w:
if layer == 3: #设置你要获取第几层
f.extend(dirnames)
break
layer += 1
(3)上述的glob方法
(4)使用pathlib方法:
import pathlib
files = [f for f in pathlib.Path().iterdir() if f.is_file()]
Path() 还附带了glob() 函数:
files = [f for f in pathlib.Path().glob("/XXX/*.log")]