aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xutils/plot3
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/plot b/utils/plot
index 2c0857b..548452b 100755
--- a/utils/plot
+++ b/utils/plot
@@ -115,6 +115,9 @@ 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'])
+ # For this script, we only want the single particle fit results
+ data = data[(data.fit_id2 == 0) & (data.fit_id3 == 0)]
+
# 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)