Vfd`dZddlmZddlZddlmZGddeZGddZdS) zgBase classes for statistical test results Created on Mon Apr 22 14:03:21 2013 Author: Josef Perktold )lzipN)Holderc>eZdZdZdfd ZdZdZdZddZxZ S) HolderTuplez Holder class with indexing Nc ttjdi||"tfd|D_dSjjf_dS)Nc38K|]}t|VdSN)getattr).0attselfs 6lib/python3.11/site-packages/statsmodels/stats/base.py z'HolderTuple.__init__..s-DDcwtS11DDDDDD)superr__init__tuple statisticpvalue)r tuple_kwds __class__s` rrzHolderTuple.__init__si)k4  )11D111  DDDDVDDDDDDJJJ.$+6DJJJrc#$K|jEd{VdSr rr s r__iter__zHolderTuple.__iter__s&:rc|j|Sr r)r idxs r __getitem__zHolderTuple.__getitem__sz#rc*t|jSr )lenrrs r__len__zHolderTuple.__len__s4:rcRtjt|j|S)N)dtype)npasarraylistr)r r%s r __array__zHolderTuple.__array__"s!z$tz**%8888rr ) __name__ __module__ __qualname____doc__rrr r#r) __classcell__)rs@rrr s77777799999999rrc8eZdZdZ d dZd dZdZdZdZdS) AllPairsResultsaResults class for pairwise comparisons, based on p-values Parameters ---------- pvals_raw : array_like, 1-D p-values from a pairwise comparison test all_pairs : list of tuples list of indices, one pair for each comparison multitest_method : str method that is used by default for p-value correction. This is used as default by the methods like if the multiple-testing method is not specified as argument. levels : {list[str], None} optional names of the levels or groups n_levels : None or int If None, then the number of levels or groups is inferred from the other arguments. It can be explicitly specified, if the inferred number is incorrect. Notes ----- This class can also be used for other pairwise comparisons, for example comparing several treatments to a control (as in Dunnet's test). hsNc||_||_|tj|dz|_n||_||_|_d|D|_dSfd|D|_dS)Ncg|]}|Srr)r pairss r z,AllPairsResults.__init__..Ns#K#K#KEE#K#K#KrcNg|]!}|dd|d"S)r-r3r)r r5levelss rr6z,AllPairsResults.__init__..PsM$<$<$<(-06eAh/?/?/?/5eAh/?/?%A$<$<$r9r=s ` rrzAllPairsResults.__init__As""  F9--1DMM$DM 0 >#K#K#K#K#KD $<$<$<$<1:$<$<$ multipletestsr:)r rAsmts rpval_correctedzAllPairsResults.pval_correctedTsL 211111111 >*F   ??BBrc*|Sr )summaryrs r__str__zAllPairsResults.__str__as||~~rc|j}tj||f}||t |j<|S)zcreate a (n_levels, n_levels) array with corrected p_values this needs to improve, similar to R pairwise output )r=r&zerosrGrr;)r k pvals_mats r pval_tablezAllPairsResults.pval_tabledsA MHaV$$ +/+>+>+@+@ $'(rc 4ddlmcm}td|jD}d|j|jz}|dd|dz dzzzd zz }|d d t|j| Dz }|S) zreturns text summarizing the results uses the default pvalue correction of the instance stored in ``self.multitest_method`` rNc34K|]}t|VdSr )r")r sss rrz*AllPairsResults.summary..vs(??BB??????rz0Corrected p-values using %s p-value correction Pairs r3z p-values  c3*K|]\}}d||fzVdS)z %s %6.4gNr)r r5pvs rrz*AllPairsResults.summary..{sGLL%;%4LLLLLLr) rBrCrDr<r?multitest_methods_namesr>joinziprG)r rFmaxleveltexts rrIzAllPairsResults.summaryos 211111111??$*>???@@D-d.CDE 3(Q,"234|CC  LLd2D4G4G4I4IJJLLLMM M r)r1NNr ) r*r+r,r-rrGrJrOrIrrrr0r0&s4?C'+<<<<& C C C C   rr0) r-statsmodels.compat.pythonrnumpyr&statsmodels.tools.testingrrr0rrrras +*****,,,,,,99999&9992WWWWWWWWWWr