dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change: less confusing wspace

Browse Source
This commit is contained in:
Benoît Sierro
2023-08-29 09:19:51 +02:00
parent 02d5b1d4ac
commit a599f55cdb
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/scgenerator/math.py
View File

@@ -187,12 +187,9 @@ def wspace(t: float | np.ndarray, t_num=0):
if isinstance(t, (np.ndarray, list, tuple)): if isinstance(t, (np.ndarray, list, tuple)):
dt = t[1] - t[0] dt = t[1] - t[0]
t_num = len(t) t_num = len(t)
t = t[-1] - t[0] + dt
else: else:
dt = t / t_num dt = t / t_num
w = 2 * pi * np.arange(t_num) / t return np.fft.fftfreq(t_num, dt) * 2 * np.pi
w = np.where(w >= pi / dt, w - 2 * pi / dt, w)
return w
def tspace(time_window: float | None = None, t_num: float | None = None, dt: float | None = None): def tspace(time_window: float | None = None, t_num: float | None = None, dt: float | None = None):