refactor: 代码格式统一和Excel导出优化
This commit is contained in:
11
main.py
11
main.py
@@ -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=1,Excel中为第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:
|
||||
|
||||
Reference in New Issue
Block a user