From ebb425f480d38102d76a720980f3a795b93f6c12 Mon Sep 17 00:00:00 2001 From: arjunsavel Date: Wed, 7 Feb 2024 15:47:06 -0500 Subject: [PATCH] find closest temperature, pressure, wl ind --- src/cortecs/eval/eval_pca.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/cortecs/eval/eval_pca.py b/src/cortecs/eval/eval_pca.py index 50e4557..47a6189 100644 --- a/src/cortecs/eval/eval_pca.py +++ b/src/cortecs/eval/eval_pca.py @@ -55,9 +55,12 @@ def eval_pca(temperature, pressure, wavelength, T, P, wl, fitter_results): The wavelength to evaluate at. """ - temperature_ind = np.where(np.isclose(T, temperature))[0][0] - pressure_ind = np.where(np.isclose(P, pressure))[0][0] - wavelength_ind = np.where(np.isclose(wl, wavelength))[0][0] + # find the nearest temperature, pressure, and wavelength indices. + + temperature_ind = np.argmin(np.abs(T - temperature)) + pressure_ind = np.argmin(np.abs(P - pressure)) + wavelength_ind = np.argmin(np.abs(wl - wavelength)) + pca_vectors, pca_coeffs_all_wl = fitter_results pca_coeffs = pca_coeffs_all_wl[wavelength_ind, :, :]