一、需求描述
两个文件夹分别存放本期数和上期数,对每个文件夹中的每个单元格对应求和汇总得出最终的结果,并写入到表格形式一样的Excel中的对应位置。
表格样式:
本期数占用区域是7-28行,上期数单元格和本期数单元格样式相同,只不过在本sheet页面的下半部分,51-72行
结果表:
二、代码实现
import os
import time
import pandas as pd
import xlrd #导入模块
from xlutils.copy import copy
#读取本期数进行汇总并写入到新的表格中
def write(this_path,excel_path,row,column):
datalist = []
dic = {}
for excel in os.listdir(this_path):
path=os.path.join(this_path,excel)
#读取本期数的数据区域
data=pd.read_excel(path,sheet_name='月报表-融资支持情况',header=row,nrows=20,usecols='A:K',index_col=0)
data=data.iloc[2:]
#重新设置表头
data.columns=(['余额','授信户数','新增授信户数','新增首次授信户数','有贷款余额户数','新发放贷款户数','新发放首次贷款户数','信用贷款余额','涉及贷款金额','支持企业家(次)数'])
datalist.append(data)
#将所有表格连接在一起
result = pd.concat(datalist, join='inner', axis=1)
#解决dataframe显示不全的问题
pd.set_option('display.max_columns', 100) # 显示全部列
pd.set_option('display.max_row', 500) # 显示全部行
pd.set_option('display.width', 1000) #解决自动换行的问题
#将求和结果和对应的行索引存入到字典中:dic[指标]=sum
index = result.columns
index1=index[:10]
for strs in index1:
#按照行进行求和
sum=result[strs].sum(axis=1)
dic[strs]=sum
#字典的取值是series类型,包含index和values两部分
#将字典转为dataframe写入到指定格式的表格中或者直接将字典数据写入到表格的指定位置中
excel_data = xlrd.open_workbook(excel_path,formatting_info=True)
wb = copy(excel_data)
table = excel_data.sheets()[0]
#series根据索引获取取值
#print(dic['余额']['大型企业'])
for c in dic.keys():
for index in dic[c].index:
# print(dic[c].index)
for i in range(row, column):
#print(table.cell(i, 0).value)
if table.cell(i, 0).value == index:
index1 = i
for j in range(0, 11):
if table.cell(row, j).value == c and len(table.cell(row, j).value) != 0:
index2 = j
value = dic[table.cell(row, index2).value][table.cell(index1, 0).value]
if value == 0 or isinstance(value, str):
continue
else:
wb.get_sheet(0).write(index1, index2, value)
if table.cell(row+1, j).value == c and len(table.cell(row+1, j).value) != 0:
index2 = j
value = dic[table.cell(row+1, index2).value][table.cell(index1, 0).value]
if value == 0 or isinstance(value, str):
continue
else:
wb.get_sheet(0).write(index1, index2, value)
#print(dic[table.cell(row+1, index2).value].values)
# wb.get_sheet(0).write(index1, index2, dic[c].values)
#print(table.cell(index1,index2).value)
wb.save(excel_path)
if __name__ == '__main__':
path = os.getcwd()
#最终写入的结果表
excel_path = os.path.join(path, '金融支持稳企业保就业情况月报表.xls')
this_path = os.path.join(path, '本期数')
last_path = os.path.join(path, '上期数')
write(this_path, excel_path, 7, 28)
print('本期数汇总结束!')
time.sleep(10)
write(last_path, excel_path, 51, 72)
input('上期数汇总结束!输入任意键结束程序!')