前言： 这个栗子本来打算是要合并自己下载的数据集，但是尝试了好久！！才发现是.xls文件损坏了…我忍…在练习过程遇到不少bug，若是小伙伴有疑问可以提一提，说不定我遇到过，恍恍惚惚…
代码：
'''
#excel_combine.py
操作：实例化Excel_Combine_tool（）,参数有2个：文件夹位置,表单名字
再save()方法
说明：代码中print()方法多为调试作用
最后文件在该源文件目录下
'''
import xlrd
import xlwt
from numpy import *
import os
class Excel_combine_tool:
file_array = []#存放多个文件名字
file_location = ''#多个文件所在的文件夹地址
n_file = 0#文件个数
sheet_name = ''#表单名字
def __init__(self,file_location,sheet_name):
self.file_location = file_location
self.sheet_name = sheet_name
#获得所有excel表格文件列表
def get_all_excel(self):
self.file_array = [fname for fname in os.listdir(self.file_location)
if os.path.isfile(os.path.join(self.file_location,fname)) and fname.endswith('.xlsx')]
print('当前文件夹有%d个.xlsx文件'%len(self.file_array))
return self.file_array
#将所有文件读到列表
def read_in(self):
self.n_file = len(self.get_all_excel())
list = []
for i in range(0,self.n_file):
complete_fname = self.file_location + '\\' + self.file_array[i]
#print(complete_fname)
table = xlrd.open_workbook(complete_fname)
try:
sheet = table.sheet_by_name(self.sheet_name)
except:
print('没有找到Sheet1,请更换表格名字')
nrows = sheet.nrows
ncols = sheet.ncols
#忽略表格第一行title
for j in range(1,nrows):
row_list = []
for k in range(0,ncols):
row_list.append(sheet.cell(j,k).value)#读取单元格数据
list.append(row_list)
#print(list)
return list
#最终保存
def save(self):
my_list = self.read_in()
file = xlwt.Workbook()
table = file.add_sheet('Sheet1')
for i in range(len(my_list)):
for j in range(len(my_list[i])):
table.write(i,j,my_list[i][j])
print('已经合并，请检查')
file.save('new.xls')
后话： 我将在另一篇博客提供python操作文件，文件夹的知识。学习输入输出会很有用的。
———关注我的公众号，一起学数据挖掘————
