From eff1bf008658ce9a77b21407bfc3175fc595a02a Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Wed, 8 Mar 2023 11:47:39 +0100 Subject: interp: allow extrapolation in interp1d --- interp.py | 1 + 1 file changed, 1 insertion(+) diff --git a/interp.py b/interp.py index e326e6d..778fc52 100644 --- a/interp.py +++ b/interp.py @@ -4,6 +4,7 @@ import numpy as np def interp1d(src_start, src_step, src_val, dst_coords, stencil): idx_src = ((dst_coords - src_start) / src_step - (stencil / 2 - 1)).astype(int) + idx_src = np.minimum(np.maximum(idx_src, 0), len(src_val) - stencil) src_coord = np.linspace(src_start, src_start + src_step * (src_val.shape[0] - 1), src_val.shape[0], dtype = src_val.dtype) -- cgit v1.2.3