linarphy/science_signal
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix edge case

Browse source
This commit is contained in:
linarphy 2024-05-27 16:34:49 +02:00
parent 9f8710c6aa
commit 4c322603d3
No known key found for this signature in database
GPG key ID: E61920135EFF2295
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/science_signal/signal.py
View file

@ -74,7 +74,7 @@ class Signal:
start = min(self.x) start = min(self.x)
if end is None: if end is None:
end = max(self.x) 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( return Signal(
self.x[indexes], self.x[indexes],
self.y[indexes], self.y[indexes],
@ -120,7 +120,7 @@ class Signal:
end = max(abs(freq_x)) end = max(abs(freq_x))
index_to_remove = where( 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 freq_y[index_to_remove] = 0
y = irfft(freq_y) y = irfft(freq_y)