获取路径下所有特定格式文件列表

情形1: 没有子目录:

方法1:

os.listdir获取当前路径下特定格式[\'.jpg\', \'png\', \'gif\']文件

def get_img_list(path):
    is_image_file = lambda x : any(x.endswith(extension) 
                                   for extension in [\'.jpg\', \'png\', \'gif\'])
    return [x for x in os.listdir(path) if is_image_file(x)]

方法2, 也可以使用glob方法,:

可以传入正则表达式r\'*.[jpg, png]\', 不能递归

glob.glob(os.path.join(path, r\'*.[jpg, png]\'))

情形2: 有子目录

使用os.walk递归获取

def file_name(file_dir):   
    L=[]   
    is_image_file = lambda x : any(x.endswith(extension) for extension in [\'.jpg\', \'png\', \'tif\'])
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if is_image_file(file):
                L.append(os.path.join(root, file))  
    return L