用于检索以特定扩展名结尾的目录中的所有文件的规范方法是什么,例如“所有以.ext
或.ext2
结尾的文件都以不区分大小写的方式结束?”一种方法是使用os.listdir
和re
模块:
import re
files = os.listdir(mydir)
# match in case-insensitive way all files that end in '.ext' or '.ext2'
p = re.compile(".ext(2)?$", re.IGNORECASE)
matching_files = [os.path.join(mydir, f) for f in files if p.search(x) is not None]
是否有一种首选方法可以更简洁地使用glob
或fnmatch
执行此操作?使用listdir
的烦恼是,必须始终通过将os.path.join
目录添加到它返回的每个文件的基本名称来处理路径。
答案 0 :(得分:3)
怎么样:
>>> import glob
>>> glob.glob("testdir/*")
['testdir/a.txt', 'testdir/b.txt', 'testdir/d.ext', 'testdir/c.ExT2']
>>> [f for f in glob.glob("testdir/*") if f.lower().endswith((".ext", ".ext2"))]
['testdir/d.ext', 'testdir/c.ExT2']