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

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

来源*

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

1. 使用glob（推荐）

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

2. 简单实用os.listdir

   import os
   for file in os.listdir("./mydir"):
       if file.endswith(".txt"):
           print(file)
   #或者使用函数式编程语句
    import os
    path = './mydir'
    text_files = [f for f in os.listdir(path) if f.endswith('.txt')]

    

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

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