python对excel指定数据提取并保存到另一excel表中(二)

改进：遍历并操作一个文件夹下所有指定格式的文件。
实现过程：用os.walk来遍历文件夹，其中会返回文件路径(dirpath)，文件夹名称(dirnames)，文件名称(filenames)，在文件名称数组中筛选文件，然后在加上所在的文件路径。

import os
import sys
from openpyxl import workbook
from openpyxl import load_workbook
#所需遍历提取excel表格的文件夹
path = r''
path1 = r''
TID = []
EPC = []
lw_save = load_workbook(path1)
sheet_save = lw_save['sheet']
start_length = sheet_save.max_row+1
#返回错误信息并结束运行
def return_err():
    print('Date Error !!!!')
    sys.exit()
for dirpath,dirnames,filenames in os.walk(path):
    for file in filenames:
        if file.endswith('.xls') or file.endswith('.xlsx'): #判断文件夹下的文件是否是excel文件
            lw = load_workbook(dirpath + '\\' + file)
            sheet = lw['RFIDData']
            if sheet.cell(1,5).value == 'EPC' and sheet.cell(1,6).value == 'TID' :#判断文件的第一行的第五列与第六列分别是否为EPC与TID
                for i in range(2,sheet.max_row+1):
                    EPC.append(sheet.cell(i,5).value)
                    TID.append(sheet.cell(i,6).value)
            else:
                return_err()
print(TID)
if TID.__len__() == EPC.__len__():
    for j in range(TID.__len__()):
        sheet_save.cell(start_length + j,1,TID[j])
        sheet_save.cell(start_length + j,2,EPC[j])
else:
    return_err()
lw_save.save(path1)