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、輸出結果

這是我手動調整過表格格式的結果。至此，需要實現的功能都實現了。