From 59e772aff8ce0f2e438622e79d9eff75b33abbc5 Mon Sep 17 00:00:00 2001
From: Raphael <raphael.sturgis@gmail.com>
Date: Thu, 18 Nov 2021 15:32:18 +0100
Subject: [PATCH] multi column operations
---
skais/ais/ais_trajectory.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py
index 7b504a0..69c9697 100644
--- a/skais/ais/ais_trajectory.py
+++ b/skais/ais/ais_trajectory.py
@@ -24,7 +24,7 @@ def compute_trajectory(times, time_gap):
def apply_func_on_window(dat, func, radius, on_edge='copy'):
- result = np.zeros(dat.shape)
+ result = np.zeros(dat.shape[0])
if on_edge == 'copy':
if len(dat.shape) == 1:
dat = np.concatenate([np.full(radius, dat[0]), dat, np.full(radius, dat[-1])])
@@ -152,4 +152,4 @@ class AISTrajectory(AISPoints):
work_df = work_df[i:]
index += i
- return result
\ No newline at end of file
+ return result
--
GitLab