對指定目錄下的指定類型文件進行遍歷,可對文件名關鍵字進行條件篩選
返回值為文件地址的列表
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('.xlsx') and 'res' 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('.'))