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