1. 程式人生 > 實用技巧 >Python遍歷目錄下檔案

Python遍歷目錄下檔案

對指定目錄下的指定型別檔案進行遍歷,可對檔名關鍵字進行條件篩選
返回值為檔案地址的列表

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('.'))