From 2bfa2b39353331d20d2582afb9294fb331b8c2eb Mon Sep 17 00:00:00 2001 From: keatonb Date: Tue, 9 Oct 2018 15:06:06 +0200 Subject: [PATCH 1/2] fix power at zero frequency --- gatspy/periodic/lomb_scargle_fast.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/gatspy/periodic/lomb_scargle_fast.py b/gatspy/periodic/lomb_scargle_fast.py index 5ae1175..6dd81fa 100644 --- a/gatspy/periodic/lomb_scargle_fast.py +++ b/gatspy/periodic/lomb_scargle_fast.py @@ -322,7 +322,11 @@ def lomb_scargle_fast(t, y, dy=1, f0=0, df=None, Nf=None, warnings.simplefilter("ignore") power = (YC * YC / CC + YS * YS / SS) / YY - + + # fix power at zero frequency + if f0 == 0: + power[0] = np.mean(y ** 2 - (y - np.mean(y)) ** 2)/np.mean(y ** 2) + # fix NaN and INF at zero frequency if np.isnan(power[0]) or np.isinf(power[0]): power[0] = 0 From a2c2850d339f33fbc53196c32e166f07e614e4ff Mon Sep 17 00:00:00 2001 From: keatonb Date: Tue, 9 Oct 2018 15:07:34 +0200 Subject: [PATCH 2/2] fix power at zero frequency --- gatspy/periodic/lomb_scargle_fast.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gatspy/periodic/lomb_scargle_fast.py b/gatspy/periodic/lomb_scargle_fast.py index 6dd81fa..46384d3 100644 --- a/gatspy/periodic/lomb_scargle_fast.py +++ b/gatspy/periodic/lomb_scargle_fast.py @@ -325,7 +325,7 @@ def lomb_scargle_fast(t, y, dy=1, f0=0, df=None, Nf=None, # fix power at zero frequency if f0 == 0: - power[0] = np.mean(y ** 2 - (y - np.mean(y)) ** 2)/np.mean(y ** 2) + power[0] = np.mean(y ** 2 - (y - np.mean(y)) ** 2) / np.mean(y ** 2) # fix NaN and INF at zero frequency if np.isnan(power[0]) or np.isinf(power[0]):