實現這一目的方法有兩種:一是,利用 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)指向的對象是否爲文件夾