python-将多个表格的信息合并到一个表格中

1、环境

代码运行环境：python3.7

相关的库：xlrd、xlwt

2、目的

通过xlrd库读取各个表格的数据，通过xlwt库将读取到的数据写入到一个表格中。

3、实现

在工程目录下，有一个test目录，存放的是待合并的表格，输出表格为merge.xls，就输出在当前工程目录下。每个合并的表格中我只需要"body"这个sheet，而且这个sheet中的数据格式都是按照下面的格式存放的。

 实现的代码如下：

# encoding: utf-8

'''
本代码的作用是将多个表格合并为一个表格。
'''

import os
import xlrd
import xlwt
import logging

# 设置logging.basicConfig()方法的参数和配置logging.basicConfig函数
FORMAT = '[%(funcName)s: %(lineno)d]: %(message)s'
LEVEL = logging.INFO
logging.basicConfig(level = LEVEL, format=FORMAT)

excel_content = []
output_file = './merge.xls'

# 打开表格，获取信息
def get_obj_list(dir_name):
    filelist = os.listdir(dir_name)
    for item in filelist :
        item = dir_name + item
        if os.path.isfile(item) and (item[-4:] == '.xls' or item[-5:] == '.xlsx' or item[-5:] == '.xlsm'):
            if item.find("$") != -1:
                continue
            merge_excel(item)
        elif os.path.isdir(item):    
            item = item + '/'
            get_obj_list(item)


# 获取单个表格的信息
def merge_excel(excelName):
    excelfd = xlrd.open_workbook(excelName)
    for sheet in excelfd.sheet_names():
        if sheet == 'body':
            print (excelName)
            sheet_content = excelfd.sheet_by_name(sheet)
            header = sheet_content.cell(0, 0).value
            if header == u'高校名称':   # 去掉标题行
                row = 1
            else:
                row = 0
            while row < sheet_content.nrows:
                college    = sheet_content.cell(row, 0).value
                institute  = sheet_content.cell(row, 1).value
                built_time = sheet_content.cell(row, 2).value
                overview   = sheet_content.cell(row, 3).value
                item = [college, institute, built_time, overview]
                excel_content.append(item)
                row += 1


# 将获取到的表格信息保存到一个表格中
def save_info():
    workbook = xlwt.Workbook(encoding = 'ascii')
    worksheet = workbook.add_sheet('merge_info')
    style = xlwt.XFStyle() # 初始化样式
    font = xlwt.Font() # 为样式创建字体
    font.name = 'Times New Roman' 
    font.bold = True # 黑体
    font.underline = True # 下划线
    font.italic = True # 斜体字
    style.font = font # 设定样式
    worksheet.write(0, 0, '高校名称')
    worksheet.write(0, 1, '学院名称')
    worksheet.write(0, 2, '成立时间')
    worksheet.write(0, 3, '人工智能学院、人工智能研究院建设情况')

    for i, item in enumerate(excel_content):
        for j in range(4):  #多添加一列（序号）
            worksheet.write(i+1, j, item[j])
    workbook.save(output_file) # 保存文件
        

if __name__ == "__main__":

    if os.path.exists(output_file):
        os.remove(output_file)

    get_obj_list('./test/')
    save_info()

4、输出结果

这是我手动调整过表格格式的结果。至此，需要实现的功能都实现了。