python 返回指定目錄下的所有檔案與特定檔案
阿新 • • 發佈:2019-02-09
有時要讀取一個資料夾下的多個檔案,下面這段程式碼即是返回指定檔案目錄下的所有檔案的檔名,存於列表中,然後從列表中進行匹配檔名中含有'.csv'的檔名,最後返回所有'.csv'檔案
讀取'.csv'檔案可以參考:
#!/usr/bin/env python # -*- coding: UTF-8 -*- import sys import os import re def eachFile(filepath): pathDir = os.listdir(filepath) #返回指定路徑下所有檔案和資料夾的名字,並存放於一個列表中 dir_list = [] for allDir in pathDir: if re.search('.csv', allDir) != None: #search()會掃描整個string查詢匹配,會掃描整個字串並返回第一個成功的匹配 dir_list.append(allDir) return dir_list if __name__ == '__main__': filepath = u"G:\\研究\\AWS\\AWS\\mod\\" #指定檔案目錄 dir_list = eachFile(filepath) #獲取檔名列表 print(dir_list)