python之xlsx合并单元格

需求背景：#

工作中将数据保存xlsx文件之后，里面每一列中有很多重复的看着很不美观，需要将每一列中的相同值合并起来，是表格看起来美观简洁#

处理前#

处理后#

直接上代码（内涵注释讲解）#

"""
-*- coding: utf-8 -*-
@Time : 2022/12/6 10:10
@Author : Logan
@File : xlxs_test.py
@Comment :  合并单元格
"""
import os
import openpyxl


def merge_xlsx(file):
    '''
    合并单元格
    :param file: 需要合并的文件
    :return:
    '''
    # 打开文件目录下的文件的第一张sheet页
    wb = openpyxl.load_workbook(os.path.join(path, file))
    sheet = wb["Sheet1"]
    print(sheet.max_row)  # 总行数
    print(sheet.max_column)  # 总列数
    # 将 所有的 列号保存下来
    column_list = []
    for i in range(1, sheet.max_column + 1):
        value = sheet.cell(row=2, column=i)
        # value.coordinate 获取当前值所在的坐标 ---> A1, B1, C1
        column_list.append(value.coordinate[0])
    print(column_list)
    # 遍历每一列 去合并
    for letter in columnList:
        sumList = []
        print(sumList)
        print(letter)
        # 先把每一列列的所有内容放入sumList中
        for i in range(2, sheet.max_row + 1):
            value = sheet.cell(row=i, column=1).value
            if value:
                sumList.append(value)
            else:
                sumList.append('')
        print(sumList)
        # 开始合并单元格
        prow = 0
        frow = 0
        flag = sumList[0]
        for i in range(len(sumList)):
            if sumList[i] != flag:
                flag = sumList[i]
                frow = i - 1
                if frow >= prow:
                    # merge_cells() ---> 合并指定坐标
                    sheet.merge_cells(f"{letter}{prow + 2}:{letter}{frow + 2}")
                    prow = frow + 1
            if i == len(sumList) - 1:
                frow = i
                sheet.merge_cells(f"{letter}{prow + 2}:{letter}{frow + 2}")

        print(f'合并完成一列： {letter}')
    # 保存表
    wb.save(os.path.join(path, file))
    print(f'操作完毕')


if __name__ == '__main__':
    # 需要操作的文件
    file_name = 'test_xlsx.xlsx'
    # 获得当前py文件的路径
    path = os.getcwd()
    print(os.path.join(path, file_name))
    merge_xlsx(file=file_name)