1. 程式人生 > 實用技巧 >Python 匯入分部分項前的 資料處理2

Python 匯入分部分項前的 資料處理2

import os
import openpyxl
from openpyxl import Workbook
from copy import deepcopy




# 原文:https://www.cnblogs.com/liuda9495/p/9039732.html




fenBuName = '二分部'
dirName = './所有分部/' + fenBuName+ '/'




#workbook2 = Workbook()
workbook2 = openpyxl.load_workbook(fenBuName + '.xlsx')# 載入excel
workbook = None
worksheet = None
unit_name = None






def is_chinese(string):
    """
    檢查整個字串是否包含中文
    :param string: 需要檢查的字串
    :return: bool
    """
    if string is None:
        return False
    for ch in string:
        if u'\u4e00' <= ch <= u'\u9fff':
            return True
    return False
    
    


def each_files():    
    pathDir =  os.listdir(dirName)
    for index, value in enumerate(pathDir):
        filepath2 = dirName + value
        print(filepath2)
        each_sheet(filepath2)




def each_sheet(path):
    global workbook
    global worksheet
    
    workbook = openpyxl.load_workbook(path)# 載入excel
    name_list = workbook.sheetnames# 所有sheet的名字    
    for sheetname in name_list:
        print(sheetname)
        worksheet = workbook[sheetname]# 讀取第一個工作表
        CellSplit()
        Business()




def CellSplit():
    
    # 獲取所有 合併單元格的 位置資訊
    # 是個可迭代物件,單個物件型別:openpyxl.worksheet.cell_range.CellRange
    # print後就是excel座標資訊
    m_list = worksheet.merged_cells

    l = deepcopy(m_list)# 深拷貝

    # 拆分合並的單元格 並填充內容
    for m_area in l:
        
        # 這裡的行和列的起始值(索引),和Excel的一樣,從1開始,並不是從0開始(注意)
        r1, r2, c1, c2 = m_area.min_row, m_area.max_row, m_area.min_col, m_area.max_col
            
        worksheet.unmerge_cells(start_row=r1, end_row=r2, start_column=c1, end_column=c2)
        print('區域:', m_area, '  座標:', r1, r2, c1, c2)
        
        # 獲取一個單元格的內容
        first_value = worksheet.cell(r1, c1).value

        # 資料填充
        for r in range(r1, r2+1):# 遍歷行        
            if c2 - c1 > 0:# 多個列,遍歷列
                for c in range(c1, c2+1):
                    worksheet.cell(r, c).value = first_value
            else:# 一個列
                worksheet.cell(r, c1).value = first_value
    




def Business():# 特定的業務邏輯
    global worksheet2
    global workbook2
    global unit_name
    
    # 獲取單位工程的名字
    unit_name = worksheet.cell(5, 1).value
    '''
    unit_name = worksheet.cell(2, 1).value
    nPos = unit_name.find(':')
    unit_name = unit_name[nPos+1:]
    '''    
    worksheet.title = unit_name# 改sheet的名字

    
    '''
    # 針對無單位工程的excel,這種情況先不考慮
    worksheet.insert_cols(1)# 插入列
    for index in range(worksheet.max_row):# 首列批量填充資料
        r = index+1
        worksheet.cell(r, 1).value = unit_name
    worksheet.cell(3, 1).value = '單位工程'# 單元格賦值
    '''

    # 刪除行
    worksheet.delete_rows(4)
    worksheet.delete_rows(2)
    worksheet.delete_rows(1)

    # 刪除列
    # worksheet.delete_cols(5)
    # worksheet.delete_cols(3)
    
    # 自動刪除列 worksheet.max_column
    for i in range(10):
        cellValue = worksheet.cell(5, 10-i).value
        ret1 = is_chinese(cellValue)
        if ret1 == False:
            worksheet.delete_cols(10-i)
    

    worksheet2 = workbook2.create_sheet(unit_name)
    for x in range(worksheet.max_row):
        r = x+1
        for y in range(worksheet.max_column):
            c = y+1
            worksheet2.cell(r, c).value = worksheet.cell(r, c).value




each_files()
workbook2.save(fenBuName+'2.xlsx')





目錄結構