使用Python脚本获取指定格式文件列表的方法

在Python环境下获取指定后缀文件列表的方式。

来源stackoverflow

这里简单以*.txt的作为例子。

1. 使用glob（推荐）

   1 import glob, os
   2 os.chdir("/mydir")
   3 for file in glob.glob("*.txt"):
   4     print(file)

2. 简单实用os.listdir
   
   

   1 import os
   2 for file in os.listdir("/mydir"):
   3     if file.endswith(".txt"):
   4         print(file)

3. 如果需要遍历目录，使用os.walk
   
   

   1 import os
   2 for root, dirs, files in os.walk("/mydir"):
   3     for file in files:
   4         if file.endswith(".txt"):
   5              print(os.path.join(root, file))