python中os.walk方法遍歷目錄的操作示例

這篇文章主要介紹了python中os.walk方法遍歷目錄的操作示例，具有一定借鑒價值，需要的朋友可以參考下。希望大家閱讀完這篇文章后大有收獲。下面讓小編帶著大家一起了解一下。

os.walk方法

#top為指定路徑
#topdown默認自上而下
os.walk(top, topdown=True, onerror=None, followlinks=False)

返回迭代器，迭代器內容：根目錄，子目錄，文件組元組;

上操作：

fpath = 'E:\imgs'
items = os.walk(fpath)
item = next(items)
print(item)

結果：

('E:\\imgs', ['2020_05', '2020_06', '2020_07', '2020_08'], ['readme.txt'])

分析：

l 第一個元素：根目錄，E:\imgs；

l 第二個元素：E:\imgs下的所有子目錄，類型為列表；

l 第三個元素：E:\imgs所有文件，類型為列表

查找圖片功能實現：

l 遍歷目錄；

l 通過文件后綴(.png/.png)判斷文件是否為圖片，

l 將文件的完成路徑保存到列表中；

代碼實現：

import os
fpath = 'E:\imgs'
#遍歷目錄
items = os.walk(fpath)
imglist = []
#遍歷items，獲取主目錄，子目錄，文件列表
for mdir, subdir, files in items:
    #遍歷文件列表
    for fname in files:
        #判斷是否為指定格式圖片
        if fname.endswith('.png') or fname.endswith('.jpg'):
            #圖片完整路徑拼接
            imgpath = os.path.join(mdir, fname)
            #添加到imglist
            imglist.append(imgpath)
#遍歷imglist
for img in imglist:
print(img)

結果：

E:\imgs\2020_05\level1.png
E:\imgs\2020_05\level2.png
...
E:\imgs\2020_08\5c4bda15af2de.jpg

感謝你能夠認真閱讀完這篇文章，希望小編分享python中os.walk方法遍歷目錄的操作示例內容對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，遇到問題就找億速云，詳細的解決方法等著你來學習!