diff options
| author | Anton Khirnov <anton@khirnov.net> | 2023-02-23 20:10:32 +0100 |
|---|---|---|
| committer | Anton Khirnov <anton@khirnov.net> | 2023-06-28 15:52:38 +0200 |
| commit | 013c9db959e118baa5ad41b35b83147d0f307385 (patch) | |
| tree | f9597f0d94d02160f247af4f12c7837b4b07eea8 | |
| parent | 69734ef9828854aa668b104d47e08526b7a2bed0 (diff) | |
doublenull: import functions from interpolate explicitly
| -rw-r--r-- | doublenull.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/doublenull.py b/doublenull.py index 78e5632..6d11bd4 100644 --- a/doublenull.py +++ b/doublenull.py @@ -1,5 +1,5 @@ import numpy as np -import scipy.interpolate as interp +from scipy.interpolate import RectBivariateSpline, interp1d from scipy.integrate import solve_ivp def calc_null_curves(times, spatial_coords, gXX, gXt, gtt, reverse = False): @@ -26,9 +26,9 @@ def calc_null_curves(times, spatial_coords, gXX, gXt, gtt, reverse = False): X=spatial_coords[i]) at time t=ray_times[j]. """ - gXX_interp = interp.RectBivariateSpline(times, spatial_coords, gXX) - gXt_interp = interp.RectBivariateSpline(times, spatial_coords, gXt) - gtt_interp = interp.RectBivariateSpline(times, spatial_coords, gtt) + gXX_interp = RectBivariateSpline(times, spatial_coords, gXX) + gXt_interp = RectBivariateSpline(times, spatial_coords, gXt) + gtt_interp = RectBivariateSpline(times, spatial_coords, gtt) if reverse: ray_times = times[::-1] @@ -109,7 +109,7 @@ def calc_null_coordinates(times, spatial_coords, u_rays, v_rays, for i, t in enumerate(times): Xu = X_of_ut[:, i] Xv = X_of_vt[:, i] - u_of_tx[i] = interp.interp1d(Xu, u_rays, fill_value = 'extrapolate')(spatial_coords) - v_of_tx[i] = interp.interp1d(Xv, v_rays, fill_value = 'extrapolate')(spatial_coords) + u_of_tx[i] = interp1d(Xu, u_rays, fill_value = 'extrapolate')(spatial_coords) + v_of_tx[i] = interp1d(Xv, v_rays, fill_value = 'extrapolate')(spatial_coords) return (u_of_tx, v_of_tx) |