B Z@s0dZddlmZmZddlZGdddeZdS)zgBase classes for statistical test results Created on Mon Apr 22 14:03:21 2013 Author: Josef Perktold )lzipzipNc@s<eZdZdZdddZdddZdd 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 : string 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 : None or list of strings 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). hsNcsj||_||_|dkr&t|d|_n||_||_|_dkrRdd|D|_nfdd|D|_dS)NcSsg|]}d|fqS)z%r).0pairsrr5lib/python3.7/site-packages/statsmodels/stats/base.py 5sz,AllPairsResults.__init__..cs(g|] }d|d|dfqS)z%s-%srrr)rr )levelsrr r 7s) pvals_raw all_pairsnpmaxn_levelsmultitest_methodr all_pairs_names)selfr rrr rr)r r __init__'s zAllPairsResults.__init__cCs4ddlmm}|dkr |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 multitestrZ multipletestsr )rrsmtrrr pval_corrected;szAllPairsResults.pval_correctedcCs|S)N)summary)rrrr __str__HszAllPairsResults.__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 )rrZzerosrrr)rkZ pvals_matrrr pval_tableKszAllPairsResults.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|VqdS)N)len)rZssrrr ^sz*AllPairsResults.summary..z0Corrected p-values using %s p-value correction ZPairs rz p-values  css|]\}}d||fVqdS)z %s %6.4gNr)rr Zpvrrr r!cs) rrrrrZmultitest_methods_namesrjoinrr)rrmaxleveltextrrr rWs zAllPairsResults.summary)rNN)N) __name__ __module__ __qualname____doc__rrrrrrrrr r s   r)r+Zstatsmodels.compat.pythonrrZnumpyrobjectrrrrr s