o 0Gf!@sFdZddlmZddlZddlmZGdddeZGdddZdS) zgBase classes for statistical test results Created on Mon Apr 22 14:03:21 2013 Author: Josef Perktold )lzipN)HoldercsDeZdZdZdfdd ZddZddZd d Zdd d ZZ S) HolderTuplez Holder class with indexing Nc sHtjdi||durtfdd|D_dSjjf_dS)Nc3s|]}t|VqdSN)getattr).0Zattself6lib/python3.10/site-packages/statsmodels/stats/base.py sz'HolderTuple.__init__..r )super__init__tupleZ statisticZpvalue)r Ztuple_kwds __class__rr rszHolderTuple.__init__ccs|jEdHdSrrrr r r __iter__szHolderTuple.__iter__cCs |j|Srr)r idxr r r __getitem__ zHolderTuple.__getitem__cCs t|jSr)lenrrr r r __len__rzHolderTuple.__len__TcCstjt|j||dS)N)dtypecopy)npZarraylistr)r rrr r r __array__!szHolderTuple.__array__r)NT) __name__ __module__ __qualname____doc__rrrrr __classcell__r r rr r src@s@eZdZdZ  dddZdddZdd Zd d Zd d ZdS)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). hsNcsl||_||_|durt|d|_n||_||_|_dur*dd|D|_dSfdd|D|_dS)NcSsg|]}d|qS)z{!r}formatrpairsr r r Ms z,AllPairsResults.__init__..cs(g|]}d|d|dqS)z{}-{}rr&r'r)levelsr r r+Qs ) pvals_raw all_pairsrmaxn_levelsmultitest_methodr-all_pairs_names)r r.r/r2r-r1r r,r r@s  zAllPairsResults.__init__cCs4ddlmm}|dur|j}|j|j|ddS)zp-values corrected for multiple testing problem This uses the default p-value correction of the instance stored in ``self.multitest_method`` if method is None. rN)methodr&)statsmodels.stats.multiteststats multitestr2Z multipletestsr.)r r4smtr r r pval_correctedWszAllPairsResults.pval_correctedcCs|Sr)summaryrr r r __str__dszAllPairsResults.__str__cCs*|j}t||f}||t|j<|S)zcreate a (n_levels, n_levels) array with corrected p_values this needs to improve, similar to R pairwise output )r1rZzerosr9rr/)r kZ pvals_matr r r pval_tablegszAllPairsResults.pval_tablecCszddlmm}tdd|jD}d|j|j}|dd|dd d 7}|d d dt|j| D7}|S) zreturns text summarizing the results uses the default pvalue correction of the instance stored in ``self.multitest_method`` rNcss|]}t|VqdSr)r)rssr r r r ysz*AllPairsResults.summary..z0Corrected p-values using %s p-value correction ZPairs r&z p-values  css |] \}}d||VqdS)z {} {:6.4g}Nr')rr*Zpvr r r r ~s) r5r6r7r0r3Zmultitest_methods_namesr2joinzipr9)r r8maxleveltextr r r r:rs   zAllPairsResults.summary)r%NNr) rr r!r"rr9r;r=r:r r r r r$%s   r$) r"Zstatsmodels.compat.pythonrZnumpyrZstatsmodels.tools.testingrrr$r r r r s