From 9cc82dad11a80b495ae19ac3a117482258303250 Mon Sep 17 00:00:00 2001 From: helene-t Date: Mon, 1 Feb 2021 16:43:00 +0100 Subject: [PATCH] [BC] fixed interpolation if subset --- SciDataTool/Functions/interpolations.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/SciDataTool/Functions/interpolations.py b/SciDataTool/Functions/interpolations.py index 1a3c2c02..13241023 100644 --- a/SciDataTool/Functions/interpolations.py +++ b/SciDataTool/Functions/interpolations.py @@ -88,17 +88,15 @@ def get_interpolation(values, axis_values, new_axis_values, index): elif isin( around(new_axis_values, 5), around(axis_values, 5) ).all(): # New axis is subset -> no interpolation - - indice_take = list() - for ii in range(len(new_axis_values)): - indice_take.append(where(axis_values == new_axis_values[ii])[0][0]) - + + indice_take = where(isin(around(axis_values, 5), around(new_axis_values, 5)))[0] + return take( values, indice_take, axis=index, ) - + else: f = interp1d(axis_values, values, axis=index, fill_value="extrapolate") return f(new_axis_values)