利用pandas
模块和glob
,之前用xlrd做过,代码很长而且只能读取xls格式,学习pandas后进行了优化,代码简短而且可以xls与xlsx一起合并
简单版本代码如下
import pandas as pd
import glob
filelocation = 'D:\\file\\' #文件夹位置
li=[]
# 读取文件夹下所有文件(xls和xlsx都读取)
for i in glob.glob(filelocation+"*.*"):
li.append(pd.read_excel(i))
# 合并后的excel文件位置
filedestination="D:\\Documents and Settings\\"
numbers = len(li)
names = 'test'
writer = pd.ExcelWriter(filedestination+names+".xlsx")
pd.concat(li).to_excel(writer,'Sheet1',index=False)
print('\n%d个文件已经合并成功!'% numbers)
writer.save()
以下为带input
的代码
import pandas as pd
import glob
#用户输入表格所在文件夹位置
filelocation = input('请输入文件夹位置(将 \ 改为 \\\\,末尾加\\\\):')
# 读取文件夹下所有文件(xls和xlsx都读取)
li=[]
for i in glob.glob(filelocation+"*.*"):
li.append(pd.read_excel(i))
# 输入 合并后的表格存放的位置 (默认桌面)
saveFile = input('\n合并后的文件是否存在桌面(y/n)?')
if saveFile == 'y':
filedestination="D:\\Documents and Settings\\Administrator\\桌面\\"
else:
filedestination = input('\n请输入合并后的文件保存位置(将 \ 改为 \\\\,末尾加\\\\):')
numbers = len(li)
names = input('\n请输入合并后的文件名称:')
writer = pd.ExcelWriter(filedestination+names+".xlsx")
pd.concat(li).to_excel(writer,'Sheet1',index=False)
print('\n%d个文件已经合并成功!'% numbers)
writer.save()