在Python環(huán)境下獲取指定后綴文件列表的方式。

來源stackoverflow

這里簡單以*.txt的作為例子。

1. 使用glob（推薦）

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

2. 簡單實用os.listdir

   1 import os2 for file in os.listdir("./mydir"):3     if file.endswith(".txt"):4         print(file)5 #或者使用函數(shù)式編程語句6  import os7  path = './mydir'8  text_files = [f for f in os.listdir(path) if f.endswith('.txt')]

    

3. 如果需要遍歷目錄，使用os.walk

   1  import os2  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))