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']
。
。
。
。
。