python实现最适合的列宽

最近因为工作需要python实现最适合的列宽

import openpyxl


# 生成列名字典，只是为了方便修改列宽时指定列，key:数字，从1开始；value:列名，从A开始
# 自适应列宽
def get_num_columns_dict():
    """
    :return: 返回字典：{1:'A', 2:'B', ...... , 52:'AZ'}
    """
    num_str_dict = {}
    a_z = [chr(a) for a in range(ord('A'), ord('Z') + 1)]
    aa_az = ['A' + chr(a) for a in range(ord('A'), ord('Z') + 1)]
    a_az = a_z + aa_az
    for i in a_az:
        num_str_dict[a_az.index(i) + 1] = i
    return num_str_dict


def style_excel(excel_name: str, sheet_name: str):
    """
    :param excel_name:
    :param sheet_name:  excel中的sheet名
    :return:
    """
    # 打开excel
    wb = openpyxl.load_workbook(excel_name)
    # 选择对应的sheet
    sheet = wb[sheet_name]
    # 获取最大行数与最大列数
    max_column = sheet.max_column
    max_row = sheet.max_row

    # 将每一列，单元格列宽最大的列宽值存到字典里，key:列的序号从1开始(与字典num_str_dic中的key对应)；value:列宽的值
    max_column_dict = {}

    # 生成列名字典，只是为了方便修改列宽时指定列，key:数字，从1开始；value:列名，从A开始
    num_str_dict = get_num_columns_dict()

    # 遍历全部列
    for i in range(1, max_column + 1):
        # 遍历每一列的全部行
        for j in range(1, max_row + 1):
            column = 0
            # 获取j行i列的值
            sheet_value = sheet.cell(row=j, column=i).value
            # 通过列表生成式生成字符列表，将当前获取到的单元格的str值的每一个字符放在一个列表中（列表中一个元素是一个字符）
            sheet_value_list = [k for k in str(sheet_value)]
            # 遍历当前单元格的字符列表
            for v in sheet_value_list:
                # 判定长度，一个数字或一个字母，单元格列宽+=1.1，其它+=2.2（长度可根据需要自行修改，经测试一个字母的列宽长度大概为1）
                if v.isdigit() is True or v.isalpha() is True:
                    column += 1.1
                else:
                    column += 2.2
            # 当前单元格列宽与字典中的对比，大于字典中的列宽值则将字典更新。如果字典没有这个key，抛出异常并将值添加到字典中
            try:
                if column > max_column_dict[i]:
                    max_column_dict[i] = column
            except Exception as e:
                print(e)
                max_column_dict[i] = column
    # 此时max_column_dict字典中已存有当前sheet的所有列的最大列宽值，直接遍历字典修改列宽
    for key, value in max_column_dict.items():
        sheet.column_dimensions[num_str_dict[key]].width = value
    # 保存
    wb.save(excel_name)


# 传入参数：Excel名称，需要设置列宽的Sheet名称
style_excel('xxx.xlsx', 'Sheet1')

复制即可使用，仅需要修改传入的Excel名及Sheet名即可