在Python环境下获取指定后缀文件列表的方式。来源stackoverflow
这里简单以*.txt的作为例子。
使用glob(推荐)1importglob, os2os.chdir("/mydir")3forfileinglob.glob("*.txt"):4PRint(file)
简单实用os.listdir1importos2forfileinos.listdir("./mydir"):3iffile.endswith(".txt"):4print(file)5#或者使用函数式编程语句6importos7path ='./mydir'8text_files = [fforfinos.listdir(path)iff.endswith('.txt')]
如果需要遍历目录,使用os.walk1importos2forroot, dirs, filesinos.walk("/mydir"):3forfileinfiles:4iffile.endswith(".txt"):5print(os.path.join(root, file))