refactor: 代码格式统一和Excel导出优化

This commit is contained in:
dmy
2026-01-04 14:01:16 +08:00
parent 06680a6e33
commit 00d480edbb
2 changed files with 254 additions and 133 deletions

11
main.py
View File

@@ -838,7 +838,8 @@ def export_all_scenarios_to_excel(results, filename):
:param filename: 输出文件路径
"""
try:
with pd.ExcelWriter(filename) as writer:
# 使用 openpyxl 引擎以便后续写入单元格
with pd.ExcelWriter(filename, engine='openpyxl') as writer:
# 1. 总览 Sheet
summary_data = []
for res in results:
@@ -882,7 +883,13 @@ def export_all_scenarios_to_excel(results, filename):
'Cost (¥)': conn['cable']['cost']
})
df = pd.DataFrame(data)
df.to_excel(writer, sheet_name=safe_name, index=False)
# 从第 2 行开始写入数据startrow=1Excel中为第2行留出第 1 行写标题
df.to_excel(writer, sheet_name=safe_name, index=False, startrow=1)
# 在第一行写入方案名称
ws = writer.sheets[safe_name]
ws.cell(row=1, column=1, value=f"Scenario: {res['name']}")
print(f"成功导出包含所有方案的Excel文件: {filename}")
except Exception as e: