Python 用 os.listdir() 获取文件列表和筛选特定格式文件

Python 用 os.listdir() 获取文件列表和过滤筛选多种格式的文件

测试代码：

folder = r"D:\aatest"
files = os.listdir(folder)
print(files)

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

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

代码说明

使用filter()函数过滤筛选出我想要的图片格式。列表的元素逐个进入过滤函数file_filter，根据返回的值确定列表元素是否留下，True留下，False过滤掉。

注意，传入的函数file_filter不能有括号

由于filter()函数在python3.x版本只是返回一个filter对象，所以要用lilst把它转化为列表格式，python2就是直接返回一个列表。

测试结果

['a.txt', 'AAA', 'BBB', 'CCC', 'DDD', '图1.jpg', '图2.png', '图3.jpg', '图四.jpg']
['图1.jpg', '图2.png', '图3.jpg', '图四.jpg']

。
。
。
。
。