import pandas as pd
from openpyxl import load_workbook
from openpyxl.utils import get_column_letter
from openpyxl.styles import Font, Alignment
def auto_width(filename='合并.xlsx'):
wb = load_workbook(filename)
for sheet in wb.sheetnames:
ws = wb[sheet]
df = pd.read_excel(filename,sheet_name=sheet, engine='openpyxl')
df.loc[len(df)]=list(df.columns)
for i in df.index:
alignment = Alignment(horizontal='center', vertical='center', text_rotation=0, wrap_text=True)
ws['A'+str(i+1)].alignment = alignment
ws['B'+str(i+1)].alignment = alignment
ws.row_dimensions[i].height = 20
for col in df.columns:
index = list(df.columns).index(col)
letter = get_column_letter(index+1)
collen = df[col].apply(lambda x :len(str(x).encode())).max()
ws.column_dimensions[letter].width = collen*0.9
wb.save(filename)