第Python實(shí)現(xiàn)遍歷讀取文件或文件夾目錄例子os.listdiros.walk遍歷讀取代碼搞機(jī)器學(xué)習(xí)或者深度學(xué)習(xí)算法很多時(shí)候需要遍歷某個(gè)目錄讀取文件，特別是經(jīng)常需要讀取某個(gè)特定后綴的文件，比如圖片的話可能需要讀取jpg,png,bmp格式的文件。python本身的庫(kù)函數(shù)功能沒(méi)有這么定制化，所以就需要再重新包裝一下。

例子

假設(shè)我們有如下的目錄結(jié)構(gòu)，以bmp結(jié)尾的是文件，其他是文件夾。下面的程序都將以該目錄結(jié)構(gòu)為例進(jìn)行說(shuō)明。

os.listdir

os.listdir僅讀取當(dāng)前路徑下的文件和文件夾，返回一個(gè)列表。讀取demo目錄結(jié)構(gòu)的代碼和結(jié)果如下：

path=r'D:\data'

items=os.listdir(path)#==['1.bmp','2.bmp','a','b']

os.walk

os.walk本身已經(jīng)是遍歷讀取，包含所有的子文件（夾）但是其結(jié)果不像是os.listdir一樣是個(gè)list，而是一個(gè)比較復(fù)雜的數(shù)據(jù)體，難以直接使用，所以一般需要再處理一下。我們可以使用for語(yǔ)句將其打印出來(lái)看看：

path=r'D:\data'

#part1

foritemsinos.walk(path):

print(items)

#part2

formain_dir,sub_dir_list,sub_file_listinos.walk(path):

print(main_dir,sub_dir_list,sub_file_list)

結(jié)果為：

(D:\\data,[a,b],[1.bmp,2.bmp])

(D:\\data\\a,[],[a1.bmp])

(D:\\data\\b,[],[b1.bmp])

D:\data[a,b][1.bmp,2.bmp]

D:\data\a[][a1.bmp]

D:\data\b[][b1.bmp]

使用迭代器對(duì)os.walk()的結(jié)果進(jìn)行輸出，發(fā)現(xiàn)每一條包含三個(gè)部分（part1），在part2中，我們給三個(gè)部分分別起名為main_dir,sub_dir_list,sub_file_list，下面對(duì)其進(jìn)行簡(jiǎn)單解釋：

main_dir：遍歷得到的路徑下所有文件夾sub_dir_list：main_dir下面的文件夾sub_file_list：main_dir下面的文件

連接main_dir和sub_file_list中的文件可以得到路徑下的所有文件。

sub_dir_list在這里則沒(méi)有用處，我們無(wú)需再去遍歷sub_dir_list，因?yàn)樗鼈円呀?jīng)包含在main_dir里了。

遍歷讀取代碼

代碼邏輯如下：

需要有后綴辨別功能，并且能夠同時(shí)辨別多個(gè)后綴

需要有遞歸和非遞歸功能

返回的是以入?yún)ath為前綴的