导出数据写入表格,格式问题还是有点费时间,好在写完了
def format(df, worksheet, max_column_width=40):
# 遍历数据框的列
for column_index, column_name in enumerate(df.columns, 1):
# 计算每列的最大长度
# 取列
selected_column = df[column_name]
# 转换列中的元素为字符串
as_strings = selected_column.astype(str)
# 计算每个字符串的长度
lengths = as_strings.map(len)
# 获取长度中的最大值
max_length = lengths.max() * 1.75
# 列名长度
len_column = len(column_name) * 2.25
# 取最大长度
max_length = max(max_length, len_column)
# 如果最大长度大于最大宽度,设置宽度为最大宽度;否则,设置为最大长度
column_width = max_length if max_length <= max_column_width else max_column_width
# 获取列字母
column_letter = openpyxl.utils.get_column_letter(column_index)
# 设置列宽(中文长度稍长一些,需乘以倍数)
worksheet.column_dimensions[column_letter].width = column_width