From 648f4f88a17550153813f95a49b5be03b9c2bfa1 Mon Sep 17 00:00:00 2001 From: Lukas Winkler Date: Thu, 1 Aug 2019 15:23:20 +0200 Subject: [PATCH] generalize rbf interpolator --- interpolators/rbf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/interpolators/rbf.py b/interpolators/rbf.py index f10da46..80b02c6 100644 --- a/interpolators/rbf.py +++ b/interpolators/rbf.py @@ -16,8 +16,9 @@ class RbfInterpolator(BaseInterpolator): if not alpha.shape: return self.rbfi(alpha, v, mcode, gamma, wt, wp) else: + size = alpha.shape[0] results = np.zeros_like(alpha) for x in range(alpha.shape[0]): for y in range(alpha.shape[0]): - results[99 - x][99 - y] = self.rbfi(alpha[0][x], v[y][0], mcode, gamma, wt, wp) + results[size - 1 - x][size - 1 - y] = self.rbfi(alpha[0][x], v[y][0], mcode, gamma, wt, wp) return results