U d»{je ã@s ddlZddd„Zd dd„ZdS) éNTécCsÞt |¡}t |¡}|jdkr&tdƒ‚|jdkr8tdƒ‚|dkrL|sLtdƒ‚|r^t |¡g}ng}td|dƒD]H}| t dtj|||¡¡| t  dtj|||¡¡qpt  |¡}|dk rÔ||}t  |¡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éN) ÚnpÚasarrayÚndimÚ ValueErrorÚ ones_likeÚrangeÚappendÚsinÚpiÚcosZvstackZ transpose)ÚtÚ frequencyÚdyÚbiasÚntermsÚcolsÚiZXT©rú~/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/timeseries/periodograms/lombscargle/implementations/mle.pyÚ design_matrixs$     "$ rc Csüttj|||fƒ\}}}|dkr,t |¡}n t |¡}t |¡}|jdkrRtdƒ‚|jdkrdtdƒ‚|jdkrvtdƒ‚|r |d}t ||¡| ¡} || }nd} t|||||d} tj   t | j | ¡t | j ||¡¡} t||||d } | t | | ¡S) aÒCompute 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) ÚmaprrrrrÚdotÚsumrZlinalgZsolveÚT) rÚyrrZt_fitZ center_dataZfit_meanrÚwZy_meanÚXZ theta_MLEZX_fitrrrÚ periodic_fit7s(       &r)NTr)TTr)ÚnumpyrrrrrrrÚs 3