diff --git a/src/science_signal/signal.py b/src/science_signal/signal.py index e0d2953..d6500d0 100644 --- a/src/science_signal/signal.py +++ b/src/science_signal/signal.py @@ -74,7 +74,7 @@ class Signal: start = min(self.x) if end is None: end = max(self.x) - indexes = where(logical_and(self.x > start, self.x < end)) + indexes = where(logical_and(self.x >= start, self.x <= end)) return Signal( self.x[indexes], self.y[indexes], @@ -120,7 +120,7 @@ class Signal: end = max(abs(freq_x)) index_to_remove = where( - logical_and(abs(freq_x) < start, abs(freq_x) > end) + logical_and(abs(freq_x) <= start, abs(freq_x) >= end) ) freq_y[index_to_remove] = 0 y = irfft(freq_y)