IR-eg ddlZddZddZdS)NTctj|}tj|}|jdkrtd|jdkrtd|dkr|std|rtj|g}ng}t d|dzD]|}|tjdtjz|z|z|z|tj dtjz|z|z|z}tj |}|||z}tj |S)aCompute the Lomb-Scargle design matrix at the given frequency. This is the matrix X such that the periodic model at the given frequency can be expressed :math:`\hat{y} = X \theta`. Parameters ---------- t : array-like, shape=(n_times,) times at which to compute the design matrix frequency : float frequency for the design matrix dy : float or array-like, optional data uncertainties: should be broadcastable with `t` bias : bool (default=True) If true, include a bias column in the matrix nterms : int (default=1) Number of Fourier terms to include in the model Returns ------- X : ndarray, shape=(n_times, n_parameters) The design matrix, where n_parameters = bool(bias) + 2 * nterms rzt should be one dimensionalrzfrequency must be a scalarz cannot have nterms=0 and no bias) npasarrayndim ValueError ones_likerangeappendsinpicosvstack transpose)t frequencydybiasntermscolsiXTs _lib/python3.11/site-packages/astropy/timeseries/periodograms/lombscargle/implementations/mle.py design_matrixrs<0 1 A 9%%Iv{{6777~5666 {{4{;<<<  Q  1fqj ! !;; BF1ru9q=94q899::: BF1ru9q=94q899:::: 4B ~ b <  cttj|||f\}}}|tj|}ntj|}tj|}|jdkrt d|jdkrt d|jdkrt d|r5|dz}tj|||z } || z }nd} t|||||} tj tj| j | tj| j ||z } t|||| } | tj| | zS) aCompute the Lomb-Scargle model fit at a given frequency. Parameters ---------- t, y, dy : float or array-like The times, observations, and uncertainties to fit frequency : float The frequency at which to compute the model t_fit : float or array-like The times at which the fit should be computed center_data : bool (default=True) If True, center the input data before applying the fit fit_mean : bool (default=True) If True, include the bias as part of the model nterms : int (default=1) The number of Fourier terms to include in the fit Returns ------- y_fit : ndarray The model fit evaluated at each value of t_fit Nrz"t, y, dy should be one dimensionalzt_fit should be one dimensionalrzfrequency should be a scalarg)rrr)rr) maprrr rr dotsumrlinalgsolveT) ryrrt_fit center_datafit_meanrwy_meanX theta_MLEX_fits r periodic_fitr-7sW."*q!Y&788OAq) z \!__ Z^^ Ju  Ev{{=>>> zQ:;;;~7888 H1' JarHHHA qsAqsAF0C0CDDI %& I I IE BF5),, ,,r)NTr)TTr)numpyrrr-rrr0sD0000f2-2-2-2-2-2-r