equency independently. This algorithm was developed by Zechmeister and Kürster which improves the Lomb-Scargle periodogram by enabling the weighting of individual samples and calculating an unknown y offset (also called a "floating-mean" model) [3]_. For more details, and practical considerations, see the excellent reference on the Lomb-Scargle periodogram [4]_. When *normalize* is False (or "power") (default) the computed periodogram is unnormalized, it takes the value ``(A**2) * N/4`` for a harmonic signal with amplitude A for sufficiently large N. Where N is the length of x or y. When *normalize* is True (or "normalize") the computed periodogram is normalized by the residuals of the data around a constant reference model (at zero). When *normalize* is "amplitude" the computed periodogram is the complex representation of the amplitude and phase. Input arrays should be 1-D of a real floating data type, which are converted into float64 arrays before processing. Parameters ---------- x : array_like Sample times. y : array_like Measurement values. Values are assumed to have a baseline of ``y = 0``. If there is a possibility of a y offset, it is recommended to set `floating_mean` to True. freqs : array_like Angular frequencies (e.g., having unit rad/s=2Ï€/s for `x` having unit s) for output periodogram. Frequencies are normally >= 0, as any peak at ``-freq`` will also exist at ``+freq``. precenter : bool, optional Pre-center measurement values by subtracting the mean, if True. This is a legacy parameter and unnecessary if `floating_mean` is True. normalize : bool | str, optional Compute normalized or complex (amplitude + phase) periodogram. Valid options are: ``False``/``"power"``, ``True``/``"normalize"``, or ``"amplitude"``. weights : array_like, optional Weights for each sample. Weights must be nonnegative. floating_mean : bool, optional Determines a y offset for each frequency independently, if True. Else the y offset is assumed to be `0`. Returns ------- pgram : array_like Lomb-Scargle periodogram. Raises ------ ValueError If any of the input arrays x, y, freqs, or weights are not 1D, or if any are zero length. Or, if the input arrays x, y, and weights do not have the same shape as each other. ValueError If any weight is < 0, or the sum of the weights is <= 0. ValueError If the normalize parameter is not one of the allowed options. See Also -------- periodogram: Power spectral density using a periodogram welch: Power spectral density by Welch's method csd: Cross spectral density by Welch's method Notes ----- The algorithm used will not automatically account for any unknown y offset, unless floating_mean is True. Therefore, for most use cases, if there is a possibility of a y offset, it is recommended to set floating_mean to True. If precenter is True, it performs the operation ``y -= y.mean()``. However, precenter is a legacy parameter, and unnecessary when floating_mean is True. Furthermore, the mean removed by precenter does not account for sample weights, nor will it correct for any bias due to consistently missing observations at peaks and/or troughs. When the normalize parameter is "amplitude", for any frequency in freqs that is below ``(2*pi)/(x.max() - x.min())``, the predicted amplitude will tend towards infinity. The concept of a "Nyquist frequency" limit (see Nyquist-Shannon sampling theorem) is not generally applicable to unevenly sampled data. Therefore, with unevenly sampled data, valid frequencies in freqs can often be much higher than expected. References ---------- .. [1] N.R. Lomb "Least-squares frequency analysis of unequally spaced data", Astrophysics and Space Science, vol 39, pp. 447-462, 1976 :doi:`10.1007/bf00648343` .. [2] J.D. Scargle "Studies in astronomical time series analysis. II - Statistical aspects of spectral analysis of unevenly spaced data", The Astrophysical Journal, vol 263, pp. 835-853, 1982 :doi:`10.1086/160554` .. [3] M. Zechmeister and M. Kürster, "The generalised Lomb-Scargle periodogram. A new formalism for the floating-mean and Keplerian periodograms," Astronomy and Astrophysics, vol. 496, pp. 577-584, 2009 :doi:`10.1051/0004-6361:200811296` .. [4] J.T. VanderPlas, "Understanding the Lomb-Scargle Periodogram," The Astrophysical Journal Supplement Series, vol. 236, no. 1, p. 16, May 2018 :doi:`10.3847/1538-4365/aab766` Examples -------- >>> import numpy as np >>> rng = np.random.default_rng() First define some input parameters for the signal: >>> A = 2. # amplitude >>> c = 2. # offset >>> w0 = 1. # rad/sec >>> nin = 150 >>> nout = 1002 Randomly generate sample times: >>> x = rng.uniform(0, 10*np.pi, nin) Plot a sine wave for the selected times: >>> y = A * np.cos(w0*x) + c Define the array of frequencies for which to compute the periodogram: >>> w = np.linspace(0.25, 10, nout) Calculate Lomb-Scargle periodogram for each of the normalize options: >>> from scipy.signal import lombscargle >>> pgram_power = lombscargle(x, y, w, normalize=False) >>> pgram_norm = lombscargle(x, y, w, normalize=True) >>> pgram_amp = lombscargle(x, y, w, normalize='amplitude') ... >>> pgram_power_f = lombscargle(x, y, w, normalize=False, floating_mean=True) >>> pgram_norm_f = lombscargle(x, y, w, normalize=True, floating_mean=True) >>> pgram_amp_f = lombscargle(x, y, w, normalize='amplitude', floating_mean=True) Now make a plot of the input data: >>> import matplotlib.pyplot as plt >>> fig, (ax_t, ax_p, ax_n, ax_a) = plt.subplots(4, 1, figsize=(5, 6)) >>> ax_t.plot(x, y, 'b+') >>> ax_t.set_xlabel('Time [s]') >>> ax_t.set_ylabel('Amplitude') Then plot the periodogram for each of the normalize options, as well as with and without floating_mean=True: >>> ax_p.plot(w, pgram_power, label='default') >>> ax_p.plot(w, pgram_power_f, label='floating_mean=True') >>> ax_p.set_xlabel('Angular frequency [rad/s]') >>> ax_p.set_ylabel('Power') >>> ax_p.legend(prop={'size': 7}) ... >>> ax_n.plot(w, pgram_norm, label='default') >>> ax_n.plot(w, pgram_norm_f, label='floating_mean=True') >>> ax_n.set_xlabel('Angular frequency [rad/s]') >>> ax_n.set_ylabel('Normalized') >>> ax_n.legend(prop={'size': 7}) ... >>> ax_a.plot(w, np.abs(pgram_amp), label='default') >>> ax_a.plot(w, np.abs(pgram_amp_f), label='floating_mean=True') >>> ax_a.set_xlabel('Angular frequency [rad/s]') >>> ax_a.set_ylabel('Amplitude') >>> ax_a.legend(prop={'size': 7}) ... >>> plt.tight_layout() >>> plt.show() ©