You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.5 KiB
68 lines
2.5 KiB
import pandas as pd
|
|
from openpyxl.styles import Font, PatternFill, Alignment, Border, Side
|
|
from openpyxl.utils import get_column_letter
|
|
|
|
def save_to_excel(data, filename):
|
|
"""
|
|
将数据集保存为格式化的Excel文件
|
|
|
|
参数：
|
|
data - 数据集 (asyncpg.Record 列表)
|
|
filename - 输出文件名 (需包含.xlsx扩展名)
|
|
"""
|
|
# 将 asyncpg.Record 转换为字典列表
|
|
data_dict = [dict(record) for record in data]
|
|
|
|
# 转换数据为DataFrame
|
|
df = pd.DataFrame(data_dict)
|
|
|
|
# 创建Excel写入对象
|
|
with pd.ExcelWriter(filename, engine='openpyxl') as writer:
|
|
df.to_excel(writer, index=False, sheet_name='Sheet1')
|
|
|
|
# 获取工作表对象
|
|
worksheet = writer.sheets['Sheet1']
|
|
|
|
# 定义边框样式
|
|
thin_border = Border(left=Side(style='thin'),
|
|
right=Side(style='thin'),
|
|
top=Side(style='thin'),
|
|
bottom=Side(style='thin'))
|
|
|
|
# 设置全局行高
|
|
for row in worksheet.iter_rows():
|
|
worksheet.row_dimensions[row[0].row].height = 20
|
|
# 为所有单元格添加边框
|
|
for cell in row:
|
|
cell.border = thin_border
|
|
|
|
# 设置标题样式
|
|
header_font = Font(bold=True, size=14)
|
|
header_fill = PatternFill(start_color='ADD8E6', end_color='ADD8E6', fill_type='solid')
|
|
|
|
for cell in worksheet[1]:
|
|
cell.font = header_font
|
|
cell.fill = header_fill
|
|
cell.alignment = Alignment(horizontal='center', vertical='center')
|
|
|
|
# 设置数据行样式
|
|
data_font = Font(size=14)
|
|
for row in worksheet.iter_rows(min_row=2):
|
|
for cell in row:
|
|
cell.font = data_font
|
|
cell.alignment = Alignment(vertical='center', wrap_text=True)
|
|
|
|
# 动态设置列宽
|
|
for idx, column in enumerate(df.columns):
|
|
# 获取列的最大长度
|
|
max_length = max(
|
|
df[column].astype(str).map(len).max(), # 数据列的最大长度
|
|
len(str(column)) # 列名的长度
|
|
)
|
|
# 计算列宽，确保在 10 到 120 之间
|
|
column_width = min(max(max_length + 2, 10) * 2, 120) # 加 2 是为了留出一些空白
|
|
# 设置列宽
|
|
column_letter = get_column_letter(idx + 1)
|
|
worksheet.column_dimensions[column_letter].width = column_width
|
|
|