From 27020846248d820d78ac2add2cb1098ea83de65d Mon Sep 17 00:00:00 2001
From: tlatorre <tlatorre@uchicago.edu>
Date: Mon, 13 Apr 2020 13:30:38 -0500
Subject: update plot script now that we have multiple fits per particle combo

---
 utils/plot | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/utils/plot b/utils/plot
index fd5ea51..2c0857b 100755
--- a/utils/plot
+++ b/utils/plot
@@ -115,6 +115,10 @@ if __name__ == '__main__':
             # merge data and fits on run and gtid
             data = data.merge(fits,left_on=['run','gtid'],right_on=['fit_run','fit_gtid'])
 
+            # Select only the best fit for a given run, gtid, and particle
+            # combo
+            data = data.sort_values('fit_fmin').groupby(['run','gtid','fit_id1','fit_id2','fit_id3'],as_index=False).nth(0).reset_index(level=0,drop=True)
+
             # calculate true kinetic energy
             mass = [SNOMAN_MASS[id] for id in data['mcgn_id'].values]
             data['T'] = data['mcgn_energy'].values - mass
-- 
cgit