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

时间:2022-10-29 12:52:51

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

来源*

这里简单以*.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)
    5 #或者使用函数式编程语句
    6  import os
    7  path = './mydir'
    8  text_files = [f for f in os.listdir(path) if f.endswith('.txt')]

     

  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))

     

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