# -*- coding: utf-8 -*-#将多个Excel文件合并成一个
import xlrd
import xlsxwriter
import os#打开一个excel文件
def open_xls(file):fh=xlrd.open_workbook(file)return fh#获取excel中所有的sheet表
def getsheet(fh):return fh.sheets()#获取sheet表的行数
def getnrows(fh,sheet):table=fh.sheets()[sheet]return table.nrows#读取文件内容并返回行内容
def getFilect(file,shnum):fh=open_xls(file)table=fh.sheets()[shnum]num=table.nrowsfor row in range(num):rdata=table.row_values(row)datavalue.append(rdata)return datavalue#添加列并返回行内容
def addcol(file,shnum):fh=open_xls(file)table=fh.sheets()[shnum]num=table.nrowsfilename = os.path.basename(file)index = filename.rfind('.')name = filename[:index]for row in range(num):acol.append(name)return acol#获取sheet表的个数
def getshnum(fh):x=0sh=getsheet(fh)for sheet in sh:x+=1return x#遍历文件夹下所有文件并输出文件名
def allxls(rootdir):allxls = []for parent, dirnames, filenames in os.walk(rootdir):for filename in filenames:if os.path.splitext(filename)[1] == '.xls':allxls.append(filename)return allxlsif __name__=='__main__':#定义要合并的excel文件列表
rootdir = "K:/7 数据分析/3 PYTHON/15 excel合并/广西"
allxls = allxls(rootdir)#存储所有读取的结果
datavalue=[]acol = []for fl in allxls:filename = os.path.basename(fl)fh=open_xls(fl)x=getshnum(fh)for shnum in range(x):print("正在读取文件："+str(fl)+"的第"+str(shnum)+"个sheet表的内容...")rvalue = getFilect(fl,shnum)acol = addcol(fl, shnum)#定义最终合并后生成的新文件
endfile = xlsxwriter.Workbook('demo.xls')#创建一个sheet工作对象
ws=endfile.add_worksheet()for a in range(len(rvalue)):for b in range(0,2):c = rvalue[a][b]d = acol[a]ws.write(a, b, c)ws.write(a, b+1, d)endfile.close()print("文件合并完成")