遍历文件夹下的所有指定类型的文件

import os
def getFiles(dir, suffix): # 查找根目录，文件后缀
    res = []
    for root, directory, files in os.walk(dir): # =>当前根,根下目录,目录下的文件
        for filename in files:
            name, suf = os.path.splitext(filename) # =>文件名,文件后缀
            if suf == suffix:
                res.append(os.path.join(root, filename)) # =>把一串字符串组合成路径
    return res

for file in getFiles('D:/Python/downtest/my_pic', '.png'): # =>查找以.py结尾的文件
    print(file)

 

os.listdir()

import os

filenames = os.listdir()
# 当前目录下的文件和文件夹
# filenames = os.listdir('.')
print(filenames)

# ['文件夹1', '文件夹2', 'filetest.py', 'order.py']