Python遍歷目錄下檔案
阿新 • • 發佈:2020-12-20
對指定目錄下的指定型別檔案進行遍歷,可對檔名關鍵字進行條件篩選
返回值為檔案地址的列表
import os # 定義一個函式,函式名字為get_all_excel,需要傳入一個目錄 def get_all_excel(dir): file_list = [] for root_dir, sub_dir, files in os.walk(r'' + dir): # 對檔案列表中的每一個檔案進行處理,如果檔名字是以‘xlxs’結尾就 # 認定為是一個excel檔案,當然這裡還可以用其他手段判斷,比如你的excel # 檔名中均包含‘res’,那麼if條件可以改寫為 for file in files: # if file.endswith('.py') and 'test' in file: if file.endswith('.xlsx'): # 此處因為要獲取檔案路徑,比如要把D:/myExcel 和res.xlsx拼接為 # D:/myExcel/res.xlsx,因此中間需要新增/。python提供了專門的 # 方法 file_name = os.path.join(root_dir, file) # 把拼接好的檔案目錄資訊新增到列表中 file_list.append(file_name) return file_list # 獲取上一級別目錄中的指定檔案 print(get_all_excel('.'))