实现这一目的方法有两种:一是,利用 os.walk() ;二是,利用 listdir() 。
1 os.walk()
利用 os.walk() 可以读取 file_dir 当前目录路径(root)、当前路径下的所有子目录(dirs)、当前路径下的所有文件(files)。
另外,利用 os.path.splitext() 可以将文件的文件名和拓展名分开,并保存在列表当中,即在列表中[0]为文件名,[1]为拓展名
import os
def file_name(file_dir):
"""读取 file_dir 目录下拓展名为'.hdf'的文件"""
L=[] #创建列表用于存储需读取的文件名
for root, dirs, files in os.walk(file_dir):
#root为所在文件夹位置,
#dirs为该目录下的文件夹
#files为该目录下的文件
for file in files:
if os.path.splitext(file)[1] == '.hdf': #将文件名和拓展名分开
L.append(os.path.join(root,file))
return L
2 listdir()
可以在下面的递归函数中体会该方法。
import os
filenames_list = []
def get_filenames(path,filenames_list):
for file in os.listdir(path):
file_path = os.path.join(path,file)
if os.path.isdir(file_path):
get_filenames(file_path,filenames_list)
else:
filenames_list.append(file)
get_filenames('H:/Mark/Mark_1',filenames_list)
for filename in filenames_list:
print(filename)
函数 get_filenames(path,filenames_list) 将 path 目录下的文件以及该路径下子目录中的文件添加到 filenames_list 列表中去。其中 listdir() 获取 path 路径下的所有文件名和文件夹名;join(path,file) 获取路径(path)下文件(file)的路径;isdir(file_path)判断路径(file_path)指向的对象是否为文件夹