Skip to content
Snippets Groups Projects
Commit ece40df6 authored by Martín Moraga's avatar Martín Moraga
Browse files

improve execution time of abs and phase functions

parent f77a3365
Branches mmo
No related tags found
No related merge requests found
...@@ -20,18 +20,14 @@ class TimeSeries: ...@@ -20,18 +20,14 @@ class TimeSeries:
def abs(self): def abs(self):
""" Calculate absolute value of complex time series. """ Calculate absolute value of complex time series.
""" """
abs_values = [] abs_values = abs(self.values)
for value in self.values:
abs_values.append(np.abs(value))
ts_abs = TimeSeries(self.name+'_abs', self.time, abs_values) ts_abs = TimeSeries(self.name+'_abs', self.time, abs_values)
return ts_abs return ts_abs
def phase(self): def phase(self):
""" Calculate phase of complex time series. """ Calculate phase of complex time series.
""" """
phase_values = [] phase_values = np.angle(self.values, deg=True)
for value in self.values:
phase_values.append(np.angle(value, deg=True))
ts_phase = TimeSeries(self.name+'_phase', self.time, phase_values) ts_phase = TimeSeries(self.name+'_phase', self.time, phase_values)
return ts_phase return ts_phase
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment