Python--利用openpyxl操作xlsx文件

最近，碰到一个简单的问题，要对excel表进行一些处理。
如下图，F和G列是合并列。当单元格内字段值中出现XX分行时，就把合并的单元格拆分开，把单元格内字段也拆分成 “ XX公司”和"xx分行"，并分别赋给拆分后的单元格。

处理前.PNG

处理后.PNG

# 利用openpyxl 库
from openpyxl import load_workbook
from openpyxl.utils import get_column_letter


def get_data(worksheet, cols, split_str):
    # 迭代每一行获取要修改的单元格的letter
    for row in range(worksheet.max_row):
        # 根据数字得到字母行列号，如第 2 列 1 行得到'B1'列，再根据此行列号获取单元格值
        # get_column_letter()函数 得到字母行列号
        col_letter1 = get_column_letter(cols) + str(row + 1)
        col_letter2 = get_column_letter(cols + 1) + str(row + 1)
        # 获取目标单元格值
        cell_value = worksheet[col_letter1].value
        # 如果单元格存在分割字符，则拆分单元格的值并分别赋值给拆分后的单元格
        if cell_value and (split_str in cell_value):
            yield cell_value, col_letter1, col_letter2


def change_excel(excel_name, sheet_name, cols, split_str, split_flags):
    try:
        # load_workbook()   打开一个excel文件
        workbook = load_workbook(excel_name)
        # workbook()  打开一个sheet
        worksheet = workbook[sheet_name]
        # 迭代每一行进行修改
        for cell_value, col_letter1, col_letter2 in get_data(worksheet, cols, split_str):
            x = cell_value.split(split_flags)
            value1 = x[0] + split_flags
            value2 = x[1]
            #  unmerge_cells()  拆分 合并单元格,然后分别赋值
            worksheet.unmerge_cells('{}:{}'.format(col_letter1, col_letter2))
            worksheet[col_letter1] = value1
            worksheet[col_letter2] = value2
        # 保存
        workbook.save('{}'.format(excel_name))
    except FileNotFoundError:
        print("未找到excel文件或sheet")
    except Exception as e:
        raise e


if __name__ == '__main__':
    excelname, sheetname = input("请输入excel文件名(例如：文件.xlsx )和sheet名，以空格分隔：").split()
    the_cols = int(input("请输入要修改的列（数字）："))
    splitstr = input("当存在什么字符时进行拆分（例：存在“分行”时进行拆分，则输入“分行”）：")
    splitflags = input("输入拆分位置（例：“上海有限公司黄浦区分行”,输入“公司”，则拆分成“上海有限公司”“黄浦区分行”）：")
    change_excel(excelname, sheetname, the_cols, splitstr, splitflags)
    # change_excel('aaa.xlsx', 'Sheet1', 2, '分行', '公司')