Skip to content

Commit

Permalink
output one combined Excel file with different sheets
Browse files Browse the repository at this point in the history
  • Loading branch information
Zhanwei-Liu committed Nov 21, 2023
1 parent 87f698b commit f674c10
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,12 +82,13 @@ def run_model(parameters, output_filename, args):
ds.to_netcdf(f'{output_filename}.nc')
logging.info("Results are written to %s.nc", output_filename)
# Write results to excel files.
for key in ds.data_vars:
if len(ds[key].shape) == 0:
df = pd.DataFrame([ds[key].values.max()], columns=[key])
else:
df = ds[key].to_dataframe()
df.to_excel(f'{output_filename}_{key}.xlsx', merge_cells=False)
with pd.ExcelWriter(f'{output_filename}.xlsx') as writer:
for key in ds.data_vars:
if len(ds[key].shape) == 0:
df = pd.DataFrame([ds[key].values.max()], columns=[key])
else:
df = ds[key].to_dataframe()
df.to_excel(writer, sheet_name=key, merge_cells=False)
logging.info("Results are written to separate excel files")


Expand Down

0 comments on commit f674c10

Please sign in to comment.