Python：借助openpyxl实现pandas to_excel自适应列宽

在使用pandas的`to_excel`方法将数据写入Excel文件时，pandas本身并没有直接提供设置自适应列宽的功能。我们可以借助openpyxl库来实现这一功能。 以下是一个示例： ```python import pandas as pd from openpyxl import Workbook from openpyxl.utils.dataframe import dataframe_to_rows # 假设df是你的DataFrame df = pd.DataFrame({ 'Column1': ['Some very long text', 'Short text'], 'Column2': [1, 2], # 更多列... }) # 创建一个ExcelWriter对象，指定引擎为openpyxl writer = pd.ExcelWriter('output.xlsx', engine='openpyxl') # 将DataFrame写入Excel df.to_excel(writer, index=False) # 获取Workbook对象 workbook = writer.book # 获取Worksheet对象，假设默认的工作表名为'Sheet1' worksheet = workbook['Sheet1'] # 遍历DataFrame的每一列，根据内容自动调整列宽 for column in df.columns: max_length = 0 for cell in dataframe_to_rows(df[column], index=False): max_length = max(max_length, len(str(cell[0]))) worksheet.column_dimensions[column].width = max_length * 1.2 # 自定义调整系数，以适应字体大小等 # 保存Excel文件 writer.save() ``` 这段代码首先计算出每列的最大长度，然后根据这个最大长度来设置Excel的列宽，实现了自适应的效果。其中`1.2`是一个调整系数，可以根据实际需要进行调整，一般用于补偿字体大小、边距等因素对列宽的影响。 ######[AI写代码神器 | 354点数解答 | 2024-01-10 17:47:52]