save txt file for each event; save pmt rates for each pmt

REINFORCE_Timeslices.py

@@ -80,8 +80,11 @@ for evt in f:
     #    #print( dom_id , rates[:5] )
     dom = det.doms[select_dom_id]
     rates = [ pmt.rate for pmt in dom.pmts ]
-    dict_pmt_rates = [pmt.id : pmt.rate for pmt in dom.pmts ]
+    dict_pmt_rates = {pmt.id : pmt.rate for pmt in dom.pmts}
     print("dict_pmt_rates", dict_pmt_rates)
+    # sort pmt rates from largest to lowest:
+    sorted_dict_pmt_rates = {k: v for k, v in sorted(dict_pmt_rates.items(), key=lambda item: item[1], reversed=True)}
+    print("sorted_dict_pmt_rates", sorted_dict_pmt_rates)
     R.append(rates[0])
     R2.append(rates[3])
     R3.append(rates[6])
@@ -149,12 +152,11 @@ while i < len(INFO_T):
 
 
 i = 0
-f = open('REINFORCE_DATA.txt', 'w')
-
-#while i < 2:#len(info_sorted):
-while i < len(info_sorted):
+while i < 2:#len(info_sorted):
+#while i < len(info_sorted):
+    f = open('REINFORCE_DATA_event_%s.txt'%i, 'w')      # write one txt file for one event
     j = 0
-    f.write('New event \n')
+    #f.write('New event \n')
     f.write(info_sorted[i][0][2])
     while j < len(info_sorted[i]):
         #print(info_sorted[i][j][0],info_sorted[i][j][1])