Python判断一个文件夹内哪些文件是图片

def is_img(ext):
    ext = ext.lower()
    if ext == '.jpg':
        return True
    elif ext == '.png':
        return True
    elif ext == '.jpeg':
        return True
    elif ext == '.bmp':
        return True
    else:
        return False

调用时

for x in os.listdir(directory):
    if is_img(osp.splitext(x)[1]):
        do something