diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py index c8b3ed0a234d9ed4186bb5c1cb234c0c327edef7..c4286bf250041bd3f11c3daa788d0ac48fd202fb 100644 --- a/skais/ais/ais_trajectory.py +++ b/skais/ais/ais_trajectory.py @@ -26,7 +26,7 @@ def compute_trajectory(times, time_gap): def apply_func_on_window(dat, func, radius, on_edge='copy'): result = np.zeros(dat.shape) if on_edge == 'copy': - dat = np.concatenate([np.full(radius, dat[0]), dat, np.full(radius, dat[-1])]) + dat = np.concatenate([np.repeat(dat[0], radius), dat, np.repeat(dat[-1], radius)]) for i in range(radius, dat.shape[0] - radius): data = dat[i - radius:i + radius + 1] result[i - radius] = func(data)