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

blueice

在Python环境下获取指定后缀文件列表的方式。
　　来源stackoverflow
　　这里简单以*.txt的作为例子。

• 使用glob（推荐）
  
  
  1 import glob, os
  2 os.chdir("/mydir")
  3 for file in glob.glob("*.txt"):
  4     print(file)
  
• 简单实用os.listdir
  
  
  1 import os
  2 for file in os.listdir("./mydir"):
  3     if file.endswith(".txt"):
  4         print(file)
  5 #或者使用函数式编程语句
  6  import os
  7  path = './mydir'
  8  text_files = [f for f in os.listdir(path) if f.endswith('.txt')]
  　　
  
  
• 如果需要遍历目录，使用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))