M/Ph%LdZddlZddlmZmZmZddlmZGddZ dS)aJ Created on Sat Oct 01 20:20:16 2011 Author: Josef Perktold License: BSD-3 TODO: check orientation, size and alpha should be increasing for interp1d, but what is alpha? can be either sf or cdf probability change it to use one consistent notation check: instead of bound checking I could use the fill-value of the interpolators N)interp1dinterp2dRbf)cache_readonlycxeZdZdZ d dZedZedZedZdZ dZ d Z d Z dS) TableDista Distribution, critical values and p-values from tables currently only 1 extra parameter, e.g. sample size Parameters ---------- alpha : array_like, 1d probabiliy in the table, could be either sf (right tail) or cdf (left tail) size : array_like, 1d The sample sizes for the table crit_table : array_like, 2d The sample sizes in the table array with critical values for sample size in rows and probability in columns asymptotic : callable, optional Callable function with the form fn(nobs) that returns len(alpha) critical values where the critical value in position i corresponds to alpha[i] min_nobs : int, optional Minimum number of observations to use the asymptotic distribution. If not provided, uses max(size). max_nobs : int, optional Maximum number of observations to use the tabular distribution. If not provided, uses max(size) Notes ----- size and alpha must be sorted and increasing. If both min_nobs and max_nobs are provided, then the critical values from the tabular distribution and the asymptotic distribution are linearly blended using the formula :math:`w cv_a + (1-w) cv_t` where the weight is :math:`w = (n - a_{min}) / (a_{max} - a_{min})`. This ensures the transition between the tabular and the asymptotic critical values is continuous. If these are not provided, then the asymptotic critical value is used for nobs > max(size). NcDtj||_|jjdkrt dtj|jdkrt dtj||_|jjdkrt dtj|jdkrt d|jjdkr9tj|dkrt dtj||_|jj |jj d|jj dfkrt dt||_ tj tj|jd |_|jdkr|jddddgf|_n|jddddgf|_d|_t#|x}|_| ||jdz}nI#t&$r<} t)| d| jd d} ~ wwxYwt|t|krt d ||_||n||_||n||_|j|krt d |j|krt d dS) Nzalpha is not 1drzalpha is not sortedzsize is not 1dzsize is not sortedz1crit_table must have shape(len(size), len(alpha))zICalling asymptotic(self.size+1) failed. The error message was: {err_msg})err_msgz,asymptotic does not return len(alpha) valueszmin_nobs > max(size)zmax_nobs > max(size))npasarrayalphandim ValueErrordiffanysize crit_tableshapelenn_alphasignmeansigncrit critv_bounds asymptoticmaxmax_size Exceptiontypeformatargsmin_nobsmax_nobs) selfrrrrr#r$rcvexcs [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/stats/tabledist.py__init__zTableDist.__init__?sZ&& :?a  .// /gdj!!Q& + + - - 4233 3Jt$$ 9>Q  -.. .gdi  A% * * , , 3122 2 9>Q  !#((** 8 !6777*Z00 ? TY_Q%79I!9L$M M M788 85zz  ; ; @ @ B BCC =1   $Aq6 :D   $Aq6 :D #&t99,4=  ! MZ  122 M M Md3ii!006sx{0K0KMMM M2ww#e**$$ "*+++(DO$,$4( $,$4( =8 # #344 4 =8 # #344 4 $ #sI J!7JJcHfdtjD}|S)NcXg|]&}tjjdd|f'SN)rrr).0ir%s r( z#TableDist.polyn..qsD///$)T_QQQT%:;;///)ranger)r%polyns` r(r2zTableDist.polynos://// --/// r0cFt|j|j|j}|Sr,)rrrr)r%poly2ds r(r4zTableDist.poly2dus $)TZAA r0c tj|jt|j\}}t |||jj d}|S)Nlinear)function) r meshgridrastypefloatrrravelrT)r%xsxapolyrbfs r(r?zTableDist.polyrbf|slTY--e44djAABbhhjj"((**do.?.E.E.G.G')))r0cL|jkr,|j|}nztdfd|jD}|jkrM|jz |j|jz z }t d|}|}||zd|z |zz}|S)a Rows of the table, linearly interpolated for given sample size Parameters ---------- n : float sample size, second parameter of the table Returns ------- critv : ndarray, 1d critical values (ppf) corresponding to a row of the table Notes ----- This is used in two step interpolation, or if we want to know the critical values for all alphas for any sample size that we can obtain through interpolation Nz?n is above max(size) and no asymptotic distribtuion is providedc&g|] }|SrB)r-pns r(r/z'TableDist._critvals..s!,,,A11Q44,,,r0g?r )rrrr2r#r$min)r%rDr&wa_cvs ` r( _critvalszTableDist._critvalss( t}  *__Q'' "<===-,,,,,,B4=  &4=4=+HIQKKq))XQ" , r0cV||}|j}|jdkr|ddd|ddd}}tj|dkrG||dkr|dS||dkr|dSt |||dS||dk}||dk}tj||}tjtj|j z}|d||<|d||<t ||||||<|S)a Find pvalues by interpolation, either cdf(x) Returns extreme probabilities, 0.001 and 0.2, for out of range Parameters ---------- x : array_like observed value, assumed to follow the distribution in the table n : float sample size, second parameter of the table Returns ------- prob : array_like This is the probability for each value of x, the p-value in underlying distribution is for a statistical test. r NrrB) rHrrr rr logical_ornanonesr) r%xrDcritvrcond_low cond_high cond_interiorprobss r(probzTableDist.probs/&q!!  =1   2;ddd 5E 71::??58||QxU2YRy )8E5))!,,R0 0E!H HU2YI]8Y???MFRWQW---E#AhE(O#BiE(O#98E5#9#9!M:J#K#KE- Lr0ctj|}|j}||}||dk}||dk}tj||}|jdkr'|rt |||StjStjtj|j z}t ||||||<|S)a Returns interpolated quantiles, similar to ppf or isf use two sequential 1d interpolation, first by n then by prob Parameters ---------- prob : array_like probabilities corresponding to the definition of table columns n : int or float sample size, second parameter of the table Returns ------- ppf : array_like critical values with same shape as prob rrJr ) r r rrHrKrrrLrMr) r%rTrDrrO cond_ilow cond_ihighrRquantiles r(critzTableDist.crits$z$ q!!E!H_ U2Y&  i<<  9>> -xu--d333v 6BGDJ///"8(5%"8"8m9L"M"Mr0c~tj|}|j}||dk}||dk}tj||}|jdkr$|r|||StjStjtj|jz}||||||<|S)a, Returns interpolated quantiles, similar to ppf or isf uses Rbf to interpolate critical values as function of `prob` and `n` Parameters ---------- prob : array_like probabilities corresponding to the definition of table columns n : int or float sample size, second parameter of the table Returns ------- ppf : array_like critical values with same shape as prob, returns nan for arguments that are outside of the table bounds rrJr ) r r rrKrr?rLrMr)r%rTrDrrVrWrRrXs r(crit3zTableDist.crit3s&z$ E!H_ U2Y&  i<<  9>> ||At,,,v 6BGDJ///"&,,q$}2E"F"Fr0)NNN) __name__ __module__ __qualname____doc__r)rr2r4r?rHrTrYr[rBr0r(rrs''R<@)-.5.5.5.5`^ ^ ^ """H,,,\%%%N&&&&&r0r) r_numpyr scipy.interpolaterrrstatsmodels.tools.decoratorsrrrBr0r(rcs  5555555555777777MMMMMMMMMMr0