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

实现过程：
一，遍历目标文件夹并筛选出’.xls’或’.xlsx’格式的表存入到数组中。
二，数组存储筛选出表的指定数据。
三，excel数据存储的起始数据为指定sheet的数据条数+1。

import os
import sys
from openpyxl import workbook
from openpyxl import load_workbook
#所需遍历提取excel表格的文件夹
path = r'xxx'
#存储到新表格
path1 = r'xxx.xlsx'
TID = []
EPC = []
lw_save = load_workbook(path1)
sheet_save = lw_save['sheet']
start_length = sheet_save.max_row+1
for file in os.listdir(path):
    if file.endswith('.xls') or file.endswith('.xlsx'): #判断文件夹下的文件是否是excel文件
        lw = load_workbook(path+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:
            print('Data Error !!!!!!!')
            sys.exit()
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:
    print('Data Error !!!!!!!')
    sys.exit()
lw_save.save(path1)