B Z.@sddZddlmZddlmZddlZddlmZm Z m Z ddl m Z Gddde Zed kr`ed d d d ddgdddZedddddddddddddddd d!d"d#d$d%d&d'geZed(d)d*d+d,d-gd.d(d/d0d1d2gd3d4d5d6d7d8gd9d:d;dgd?d@dAdBdCdDgdEdFd9dGdHdIgdJdKdLdMd>sz#TableDist.polyn..)rr)rpolynr)rrr<s zTableDist.polyncCst|j|j|j}|S)N)rr r r)rpoly2drrrrBszTableDist.poly2dcCs>t|jt|j\}}t|||jj dd}|S)NZlinear)Zfunction) r Zmeshgridr Zastypefloatr rZravelrT)rZxsZxapolyrbfrrrrIs zTableDist.polyrbfcstfdd|jDS)arows 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 csg|] }|qSrr)rp)nrrrcsz'TableDist._critvals..)r arrayr)rrr)rr _critvalsOszTableDist._critvalsc Cs||}|j}|jdkr8|ddd|ddd}}t|dkr||dkrZ|dS||dkrn|dSt|||dS||dk}||dk}t||}tjt|j }|d||<|d||<t||||||<|SdS)afind pvalues by interpolation, eiter cdf(x) or sf(x) returns extrem 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 : arraylike This is the probability for each value of x, the p-value in underlying distribution is for a statistical test. r Nrr) r!r rr r r logical_ornanonesshape) rxrcritvr Zcond_lowZ cond_high cond_interiorZprobsrrrprobes$        zTableDist.probcCs |||S)zreturns interpolated quantiles, similar to ppf or isf this can be either cdf or sf depending on the table, twosided? this doesn't work, no more knots warning )r)rr*rrrrcrit2szTableDist.crit2c Cst|}|j}||}||dk}||dk}t||}|jdkr`|rZt|||StjStjt|j }t||||||<|S)areturns 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 rr"r ) r r r r!r#r rr$r%r&) rr*rr r( cond_ilow cond_ihighr)quantilerrrcrits      zTableDist.critcCs~t|}|j}||dk}||dk}t||}|jdkrT|rN|||StjStjt|j}||||||<|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 rr"r ) r r r r#r rr$r%r&)rr*rr r,r-r)r.rrrcrit3s      zTableDist.crit3N)__name__ __module__ __qualname____doc__rrrrrr!r*r+r/r0rrrrrs   9 'r__main__g?g333333?g?g?g{Gz?gMbP?r" (diii/iAiZixiii!i?iWiii ii)iCisiiii0i_ii i iMiii=imii`i#iRiEii:i1i(iiii isxJMRYhz%')-4=#*g@@gsh|??z should be:z interp2d badztwo 1drg(\?z should be in:)g'1Z?gsh|??gRQ?gFx?cCs*g|]"}t|D]}t|t|qqSr)crit_lflfr*r )rrcrrrr5srcCs(g|] }tjD]}t|t|qqSr)rr r/r )rrrrrrr7scCs(g|] }tjD]}t|t|qqSr)rr r0r )rrrrrrr9scCs4g|],}tjdddD]}t|t|qqS)Nr"g?)rr r0r )rrrrrrr;sr cCs4g|],}tjdddD]}t|t|qqS)Nr"g?)rr r0r )rrrrrrr<scCs6g|].}t|ddfD]}t|dt|qqS)Nr"g?)rrr*r )rrrrrrr=scCs6g|].}t|ddfD]}t|dt|qqS)r Ng?)rrr*r )rrrrrrr>scCs6g|].}t|ddfD]}t|t|dqqS)Nr"g?)rrr*r )rrrrrrr@s)!r4Z __future__rZstatsmodels.compat.pythonrZnumpyr Zscipy.interpolaterrrZstatsmodels.tools.decoratorsrobjectrr1r r rr rrprintr*r+r/r0maxabsrZreshaperallrrrrsx   b * <::FF@@