Skip to content
Snippets Groups Projects
Commit a9e002df authored by Feifei Huang's avatar Feifei Huang
Browse files

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

parent a863edde
No related branches found
No related tags found
Loading
......@@ -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])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment