From d6f38aad17f06e19d1e143db485fb12e96817f29 Mon Sep 17 00:00:00 2001 From: AbbyGi Date: Fri, 9 Aug 2019 15:21:18 -0400 Subject: [PATCH] Fixed cut data function --- xrdphase/phase_id_functions.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/xrdphase/phase_id_functions.py b/xrdphase/phase_id_functions.py index 87556cd..5338078 100644 --- a/xrdphase/phase_id_functions.py +++ b/xrdphase/phase_id_functions.py @@ -84,16 +84,9 @@ def find_nearest(array, value): def cut_data(qt, sqt, qmin, qmax): - qcut = [] - sqcut = [] - for i in range(len(qt)): - if qt[i] >= qmin and qt[i] <= qmax: - qcut.append(qt[i]) - sqcut.append(sqt[i]) - - qcut = np.array(qcut) - sqcut = np.array(sqcut) - return qcut, sqcut + qt_back, sqt_back = qt[qt > qmin], sqt[qt > qmin] + qt_back, sqt_back = qt_back[qt_back < qmax], sqt_back[qt_back < qmax] + return qt_back, sqt_back def read_index_data_smart(filename, junk=None, backjunk=None, splitchar=None,