功能

获取文件夹的的文件+获取文件夹里的特定格式的文件，比如.png等

程序

import  os
folder = r"D:\python\untitled1"
files = os.listdir(folder)
print(files)

def file_filter(f):
    if f[-4:] in ['.jpg', '.png']:
        return True
    else:
        return False

files = list(filter(file_filter, files))
print(files)

效果

想要查看的文件夹

运行程序

可以看到，首先打印出所有格式的文件名，之后打印我们锁定的格式的文件名。