Python+Xlwings操作Excel刪除行和列


一、需求:

  某公司管理的多個資管計划每天生成A表,業務人員需手工打開每個A表,將某些行、列刪除后方可打印上報。

  現擬采用程序代替手工操作。

 

二、分析:

  1、應在原始文件的副本上操作,因此需拷貝文件夾內所有Excel至目標目錄;

  解答:使用shutil.copy()

 

  2、需打開excel並刪除指定的行和列;

  解答:openpyxl不支持xls格式,xlwt無法刪除行和列,最終選擇xlwings;

 

三、代碼實現:

 

#!/usr/bin/env python
# _*_ coding:utf-8 _*_

"""

@Time    : 2019-12-27 17:16
@Author  : Peanut_C
@FileName: excel_converter.py

"""


import os
import shutil
import xlwings as xw

current_dir = os.getcwd()
src_dir = os.path.join(current_dir, 'src_dir')
dst_dir = os.path.join(current_dir, 'dst_dir')
exist_list = ['YYYY', 'XXXX']  # 要保留行的A列關鍵字


def file_copy(source_dir, destination_dir):
    os.chdir(source_dir)
    for file in os.listdir(source_dir):
        shutil.copy(file, destination_dir)
    print('INFO ===>>> 文件拷貝完成!')


def excel_modifier(wk_dir):
    os.chdir(wk_dir)
    for file in os.listdir(wk_dir):
        # 檢查文件格式是否為xls
        # print(type(os.path.splitext(file)[1]))
        if os.path.splitext(file)[1] != '.xls':
            print(file, '===>>>文件格式不正確,請檢查!')
        else:
            print('開始處理===>>>', file)
            # 創建app,打開工作表
            app = xw.App(visible=False, add_book=False)
            app.screen_updating = False
            app.display_alerts = False
            load_wb = app.books.open(file)
            load_ws = load_wb.sheets.active
            print('\t已打開工作表……')

            # 獲取總行數(列數固定不需要獲取)
            rows = load_ws.api.UsedRange.Rows.count
            # cols = load_ws.api.UsedRange.Columns.count

            # 獲取需要處理的A列范圍
            a_range = load_ws.range('A1:A'+str(rows-4))  # 得到range對象

            # 將range中每行對象存放到列表中並倒序
            print('\t開始獲取標志列……')
            cell_list = []
            for cell in a_range:
                cell_list.append(cell)
            cell_list.reverse()
            # print(cell_list)

            # 將表頭拆分、重新合並,為插入的值騰地方
            print('\t開始調整合並單元格……')
            load_ws.range('H3:J3').api.unmerge()  # 拆分單元格
            load_ws.range('H3:I3').api.merge()  # 合並單元格
            load_ws.range('J3').value = 'xxx'  # 插入值

            # 設定將A列每個值與要保留列表比對,比對不上則刪除整行
            print('\t開始調整行和列……')
            for cell in cell_list:
                if cell.value is not None:  # 單元格不為空則開始比對
                    find_flag = 0  # 匹配標志
                    for exist_value in exist_list:
                        if cell.value.find(exist_value) != -1:
                            find_flag = 1  # 匹配則將標志置為1
                            break  # 一個單元格只要匹配就不再比對保留列表剩下的值
                        else:
                            continue  # 匹配不上則繼續
                    if find_flag == 0:  # 沒匹配上的刪除整行
                        cell_to_del = cell.address
                        # print(cell_to_del)
                        load_ws.range(cell_to_del).api.EntireRow.Delete()
                else:  # 單元格為空直接刪除
                    cell_to_del = cell.address
                    # print(cell_to_del)
                    load_ws.range(cell_to_del).api.EntireRow.Delete()

            # 處理列,將指定列從大到小刪除(避免先刪除小列導致后續列號變動)
            load_ws.api.columns('K').delete
            load_ws.api.columns('G').delete
            load_ws.api.columns('B').delete
            # 美化處理后的Excel
            print('\t開始美化表格……')
            load_ws.range('A1:H24').columns.autofit()
            # 處理完畢,保存、關閉、退出Excel
            load_wb.save()
            load_wb.close()
            app.quit()
            print('處理完畢===>>>', file, '\n\n')


if __name__ == '__main__':
    file_copy(src_dir, dst_dir)
    excel_modifier(dst_dir)
    print('任務結束,請至dst_dir目錄查看文件!\n\n')
    os.system('pause')

 

 

 

 

四、運行情況:

  腳本測試完畢后,使用pyinstaller -F excel_converter.py -i icon.ico打包成為exe文件。

  將可執行程序拷貝至業務人員電腦可直接執行,原始文件拖入src_dir,處理后文件輸出至dst_dir。

  經測試excel2013使用正常,excel2007無法連接。

  

  希望能幫到有需要的朋友。

  多多指教!

 

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM