From ece40df65bcf664e192fedd7bea9b417ee5fd9e9 Mon Sep 17 00:00:00 2001
From: Martin Moraga <martin.moraga@eonerc.rwth-aachen.de>
Date: Mon, 13 Feb 2023 09:02:02 +0100
Subject: [PATCH] improve execution time of abs and phase functions
Signed-off-by: Martin Moraga <martin.moraga@eonerc.rwth-aachen.de>
---
villas/dataprocessing/timeseries.py | 8 ++------
1 file changed, 2 insertions(+), 6 deletions(-)
diff --git a/villas/dataprocessing/timeseries.py b/villas/dataprocessing/timeseries.py
index 5610211..9e35359 100644
--- a/villas/dataprocessing/timeseries.py
+++ b/villas/dataprocessing/timeseries.py
@@ -20,18 +20,14 @@ class TimeSeries:
def abs(self):
""" Calculate absolute value of complex time series.
"""
- abs_values = []
- for value in self.values:
- abs_values.append(np.abs(value))
+ abs_values = abs(self.values)
ts_abs = TimeSeries(self.name+'_abs', self.time, abs_values)
return ts_abs
def phase(self):
""" Calculate phase of complex time series.
"""
- phase_values = []
- for value in self.values:
- phase_values.append(np.angle(value, deg=True))
+ phase_values = np.angle(self.values, deg=True)
ts_phase = TimeSeries(self.name+'_phase', self.time, phase_values)
return ts_phase
--
GitLab