From 29df63e3787deb7fa81dad6bfcfb276f129601e5 Mon Sep 17 00:00:00 2001
From: Raphael <raphael.sturgis@gmail.com>
Date: Sat, 6 Nov 2021 10:47:21 +0100
Subject: [PATCH] random
---
skais/ais/ais_trajectory.py | 9 ++++-----
1 file changed, 4 insertions(+), 5 deletions(-)
diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py
index 4185386..157e2f0 100644
--- a/skais/ais/ais_trajectory.py
+++ b/skais/ais/ais_trajectory.py
@@ -29,7 +29,7 @@ def compute_trajectory(times, time_gap, size_limit):
return i
-@jit(nopython=True)
+#@jit(nopython=True)
def compute_std(dat, radius):
stds = np.empty(dat.shape[0])
@@ -173,13 +173,14 @@ def l2_angle(dat, radius):
def angle_dispersion(dat, radius):
- l2 = np.zeros(dat.shape)
+ disp = np.zeros(dat.shape)
dat = np.concatenate([np.full(radius, dat[0]), dat, np.full(radius, dat[-1])])
for i in range(radius, dat.shape[0] - radius):
data = dat[i - radius:i + radius + 1]
- l2[i - radius] = angular_dispersion(np.radians(data))
+ disp[i - radius] = angular_dispersion(np.radians(data))
+ return disp
class AISTrajectory:
def __init__(self, df, interpolation_time=None):
@@ -283,8 +284,6 @@ class AISTrajectory:
if field in self.df.columns:
dat = self.df[field].to_numpy()
disp = angle_dispersion(dat, radius)
- disp[-radius:] = np.nan
- disp[:radius] = np.nan
self.df[f"{field}_disp"] = disp
def compute_position_features(self, radius):
--
GitLab