Python-目录下相同格式的Excel文件合并
最近在客户现场接到一个任务,需要将全国所有省份的数据进行合并。目录是分层级的,首先是省份目录、然后地级市目录、最里面是区县目录。
需要将每个目录中的数据进行合并,然后添加4列数据,并将某一个列的数据进行拆分和处理。总共将近3000张个Excel文件需要处理,
如果手工处理的话,估计要搞一个星期才能搞完。为了在规定时间内完成任务,我通过编写python程序实现了这样的功能。
这里我简单分享下单个目录中Excel文件合并的模块。
Python代码:
#!/usr/bin/env python
# coding:utf-8
"""
@File Name: zwc002.py
@Version: 1.0
@Python Version: 3.7
@Author: liguanbin
@Created Time: 2021/5/21 9:04
@Software: PyCharm
@Desc:
"""
import os
import xlrd
import xlsxwriter
import glob
# 获取要合并的所有excel表格
def get_excel():
global file_path
global current_folder
global new_excel_path
file_path = input("请输入Excel文件所在的目录:")
if not file_path.endswith('\\'):
current_folder = os.path.basename(file_path)
file_path = file_path + "\\"
else:
current_folder = os.path.basename(os.path.dirname(file_path))
new_excel=current_folder + "_合并结果.xlsx"
new_excel_path = os.path.join(file_path,new_excel)
# 新建的exce文件绝对路径
if os.path.isfile(new_excel_path):
os.remove(new_excel_path)
# 如果文件已经存在就删除,避免重复执行插入重复
all_excel = glob.glob(file_path + "*.xlsx")
print("该目录下有" + str(len(all_excel)) + "个Excel文件:")
if (len(all_excel) == 0):
return 0
else:
for i in range(len(all_excel)):
print(all_excel[i])
return all_excel
# 获取sheet下的数据
def get_sheet_data(sheet):
row = sheet.nrows
for i in range(row):
if (i == 0):
global biao_tou
biao_tou = sheet.row_values(i)
continue
values = sheet.row_values(i)
all_data1.append(values)
return all_data1
if __name__ == '__main__':
all_excel = get_excel()
# 得到要合并的所有exce表格数据
if (all_excel == 0):
print("该目录下无.xlsx文件!请检查您输入的目录是否有误!")
exit()
all_data1 = []
# 用于保存合并的所有行的数据
# 下面开始文件数据的获取
for excel in all_excel:
fh = xlrd.open_workbook(excel)
# 打开文件
sheets = fh.sheets()
# 获取文件下的所有sheet
for sheet_index in range(len(sheets)):
all_data1 = get_sheet_data(sheets[sheet_index])
# 获取一个sheet下的所有行的数据,追加到数组中
all_data1.insert(0, biao_tou)
# 表头写入(只需要写入一个)
fh1 = xlsxwriter.Workbook(new_excel_path)
# 新建一个excel表
title_formater=fh1.add_format({'border': 1,'bold':True,'align':'center','bg_color':'3399FF','font_size':12})
# 标题格式:边框:1、加粗、居中、填充:蓝色、字体:12
common_formater = fh1.add_format({'border': 1})
# 数据格式:边框:1
new_sheet_name="执常委列表"
# 指定sheet的名称
new_sheet = fh1.add_worksheet(new_sheet_name)
# 新建一个sheet
for i in range(len(all_data1)):
for j in range(len(all_data1[i])):
c = all_data1[i][j]
if i==0:
new_sheet.write(i, j, c, title_formater)
else:
new_sheet.write(i, j, c,common_formater)
fh1.close()
# 关闭该exce表
print("\033[1;34m文件合并成功,请查看" + new_excel_path + "\033[0m")
效果演示:
合并后的文件: