python获取文件夹里有什么文件+查看特定格式的文件

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)