功能
其中Path代表文件夾的路徑 ,layer代表讀取文件夾的深度 ,例如0代表要讀取的深度中沒有次級文件夾,1代表有一個次級文件夾,注意讀取的是存放在layer深度的所有文件,淺一級的深度不做讀取。
代碼
import os
def readFileDocument(path,layer):#採用遞歸的方法讀取文件
layerN =layer
fileDocument = os.listdir(path)
subDocument = []
for document in fileDocument:
if layerN>0:
#如果是文件夾繼續讀取
if (os.path.isdir(path + '/' + document, )):
subDocument.append((readFileDocument(path+'/'+document,layerN-1)))
else:
subDocument.append(path + '/' + document)
return subDocument
#使用範例,讀取深度爲1的所有文件名
path=r'F:\BaiduNetdiskDownload\DATA\train'
location = readFileDocument(path,1)
返回的值
返回的值爲一個(layer+1)維列表 ,列表中的值爲文件的絕對路徑
讀取方法爲雙重循環讀取,如(接上題讀取layer=1的二維列表):
for document in location:
for name in doucument:
print(name)#輸出絕對路徑