python合並目錄下excel數據

合並同級目錄下所有excel文件

import os
import datetime
import xlrd
import xlsxwriter

# 遍歷的文件目錄
file_path = './test'
# 寫入的目標文件：
tar_file = file_path + '/mixed.xlsx'


# 遍歷目錄下的所有文件名
def all_path(dirname):
    result = []
    # 當前主目錄,當前主目錄下的所有目錄,當前主目錄下的所有文件
    for maindir, subdir, file_name_list in os.walk(dirname):
        for filename in file_name_list:
            if filename.endswith('xls') or filename.endswith('xlsx'):
                apath = os.path.join(maindir, filename)  # 合並成一個完整路徑
                result.append(apath)

    return result


# 獲取excel文件的內容數據
def concat_and_insert(fdir):
    records = []
    if len(fdir) > 0:
        for dir in fdir:
            # 讀文件
            data = xlrd.open_workbook(dir)
            # 第一個sheet頁的名稱;
            first_sheet = data.sheet_by_index(0).name
            print(dir, '>' * 10, first_sheet)

            # 獲取sheet頁的名稱
            sheet = data.sheet_by_name(first_sheet)
            # 獲取表的行數：
            nrows = sheet.nrows
            for i in range(nrows):
                # 跳過第一行
                if i < 5:
                    continue
                records.append(sheet.row_values(i))
    return records


def insert_file(alist, tarfile):
    # 新建目標文件
    wh = xlsxwriter.Workbook(tarfile)
    wadd = wh.add_worksheet('total')
    if len(alist) > 0:
        for row_num, row_data in enumerate(alist):
            wadd.write_row(row_num + 1, 0, row_data)
    wh.close()


if __name__ == "__main__":
    strat = datetime.datetime.now()
    print(strat)
    filename = all_path(file_path)
    print(filename)

    records = concat_and_insert(filename)

    # 寫入文件
    insert_file(records, tar_file)

    end = datetime.datetime.now()
    print(end)
    print("持續時間{}".format(end - strat))
    print('ok')