p7]c@s dZddlmZddlmZddlZddlmZm Z m Z ddl m Z de fdYZed kr ejd d d d ddgdddZejddddddddddddddddd d!d"d#d$d%d&geZejd'd(d)d*d+d,gd-d'd.d/d0d1gd2d3d4d5d6d7gd8d9d:d;d<d=gd>d?d@dAdBdCgdDdEd8dFdGdHgdIdJdKdLd;dMgdNdOdPdQdRdSgdTdUdVdWd3dXgdYdZd[d\d]d^gd_d`dNdadLdbgdcdddedfdgdhgdidjdkdldmdngdodpdddqdrdsgdtdudvdZd\dwgdxdydcdzdJd{gd|d}did~ddgddd|ddedgdddd|dvdIgddddddgddddddgddddddgd!ddd"ddggdddddfdZeeeeZeejdd dd edeejd d dddeejd d dddeejd d dddedeejdd dddeejdd dddeejdd dddedeejdd dddeejdd dddeejdd dddedeejdd dddeejdd dddeejdd dddeejejejgeeeD]-ZeeD]Zejeee^qqj dej!ejeejejejgeeeD],ZejD]Zejeee^qqj dej!eeejejejgeeeD],ZejD]Zejeee^qqj dej!eeejgeeeD]4Zejd dD]Zejeee^qxqcj dej!deddddfkj"eejgeeeD]4Zejd dD]Zejeee^qqj dej!deddddfkj"eejgeeeD]=ZeeddfD] Zejedee^qqj dej!dejd kj"eejgeeeD]=ZeeddfD] Zejedee^q$q j dej!dejdkj"eejgedeeD]=ZeeddfD] Zejeeed^qqj dej!dejd kj"ndS(sM 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 i(tprint_function(trangeN(tinterp1dtinterp2dtRbf(tcache_readonlyt TableDistcBsqeZdZdZedZedZedZdZdZ dZ dZ d Z RS( sDistribution, 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 second paramater in the table crit_table : array_like, 2d array with critical values for sample size in rows and probability in columns Notes ----- size and alpha should be increasing cCstj||_tj||_tj||_t||_tjtj|jdj |_ |j dkr|jddddgf|_ n"|jddddgf|_ dS(Nii( tnptasarraytalphatsizet crit_tabletlentn_alphatsigntdifftmeantsigncritt critv_bounds(tselfR R R ((s:lib/python2.7/site-packages/statsmodels/stats/tabledist.pyt__init__0s'%cCsEgt|jD]+}t|j|jdd|f^q}|S(N(RR RR R (Rtitpolyn((s:lib/python2.7/site-packages/statsmodels/stats/tabledist.pyR<s>cCst|j|j|j}|S(N(RR R R (Rtpoly2d((s:lib/python2.7/site-packages/statsmodels/stats/tabledist.pyRBscCs[tj|jjt|j\}}t|j|j|jj jdd}|S(Ntfunctiontlinear( RtmeshgridR tastypetfloatR RtravelR tT(Rtxstxatpolyrbf((s:lib/python2.7/site-packages/statsmodels/stats/tabledist.pyR!Is'0cCs)tjg|jD]}||^qS(srows 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 (RtarrayR(Rtntp((s:lib/python2.7/site-packages/statsmodels/stats/tabledist.pyt _critvalsOsc Cs:|j|}|j}|jdkrQ|ddd|ddd}}ntj|dkr||dkr~|dS||dkr|dSt|||dS||dk}||dk}tj||}tjtj|j }|d||<|d||sv  .*-;    {ww