# HG changeset patch
# User luca_milaz
# Date 1720178917 0
# Node ID 0d1ae4dffac7b00337705255ea302a8f25c7c4a1
# Parent  f3ade2c90a3de7c3a19a5944af16fcdbbe63f2ae
Uploaded

diff -r f3ade2c90a3d -r 0d1ae4dffac7 marea_2_0/flux_sampling.py
--- 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
         
 ##############################################################################