changeset 209:0d1ae4dffac7 draft

Uploaded
author luca_milaz
date Fri, 05 Jul 2024 11:28:37 +0000
parents f3ade2c90a3d
children eca31a79e4b6
files marea_2_0/flux_sampling.py
diffstat 1 files changed, 5 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/marea_2_0/flux_sampling.py	Fri Jul 05 11:21:47 2024 +0000
+++ b/marea_2_0/flux_sampling.py	Fri Jul 05 11:28:37 2024 +0000
@@ -94,8 +94,8 @@
     print(s)
 
 
-def write_to_file(dataset: pd.DataFrame, name: str)->None:
-    dataset.to_csv(ARGS.output_folder + name + ".csv", sep = '\t', index = False)
+def write_to_file(dataset: pd.DataFrame, name: str, keep_index:bool=False)->None:
+    dataset.to_csv(ARGS.output_folder + name + ".csv", sep = '\t', index = keep_index)
 
 
 
@@ -275,17 +275,17 @@
     if("mean" in ARGS.output_types):
         all_mean = all_mean.fillna(0.0)
         all_mean = all_mean.sort_index()
-        write_to_file(all_mean, "mean")
+        write_to_file(all_mean, "mean", True)
 
     if("median" in ARGS.output_types):
         all_median = all_median.fillna(0.0)
         all_median = all_median.sort_index()
-        write_to_file(all_median, "median")
+        write_to_file(all_median, "median", True)
     
     if("quantiles" in ARGS.output_types):
         all_quantiles = all_quantiles.fillna(0.0)
         all_quantiles = all_quantiles.sort_index()
-        write_to_file(all_quantiles, "quantiles")
+        write_to_file(all_quantiles, "quantiles", True)
     pass
         
 ##############################################################################