U Kv f@sFdZddlmZddlZddlmZGdddeZGdddZdS) zgBase classes for statistical test results Created on Mon Apr 22 14:03:21 2013 Author: Josef Perktold )lzipN)HoldercsDeZdZdZd fdd ZddZddZd d Zdd d ZZ S) HolderTuplez Holder class with indexing Nc sFttjf||dk r4tfdd|D_njjf_dS)Nc3s|]}t|VqdSN)getattr).0Zattself:/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/stats/base.py sz'HolderTuple.__init__..)superr__init__tupleZ statisticZpvalue)r tuple_kwds __class__rr rszHolderTuple.__init__ccs|jEdHdSrrrr r r __iter__szHolderTuple.__iter__cCs |j|Srr)r idxr r r __getitem__szHolderTuple.__getitem__cCs t|jSr)lenrrr r r __len__szHolderTuple.__len__cCstjt|j|dS)N)dtype)npZasarraylistr)r rr r r __array__"szHolderTuple.__array__)N)N) __name__ __module__ __qualname____doc__rrrrr __classcell__r r rr r s rc@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 : 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). hsNcsj||_||_|dkr&t|d|_n||_||_|_dkrRdd|D|_nfdd|D|_dS)NcSsg|]}d|fqS)z%rr rpairsr r r Nsz,AllPairsResults.__init__..cs(g|] }d|d|dfqS)z%s-%srr%r r&levelsr r r(Ps  ) pvals_raw all_pairsrmaxn_levelsmultitest_methodr*all_pairs_names)r r+r,r/r*r.r r)r rAs 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 multitestr/Z multipletestsr+)r r1smtr r r pval_correctedTszAllPairsResults.pval_correctedcCs|Sr)summaryrr r r __str__aszAllPairsResults.__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 )r.rzerosr6rr,)r kZ pvals_matr r r pval_tabledszAllPairsResults.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 vsz*AllPairsResults.summary..z0Corrected p-values using %s p-value correction ZPairs r%z p-values  css|]\}}d||fVqdS)z %s %6.4gNr )rr'pvr r r r {s) r2r3r4r-r0Zmultitest_methods_namesr/joinzipr6)r r5maxleveltextr r r r7os   zAllPairsResults.summary)r$NN)N) rrr r!rr6r8r;r7r r r r r#&s   r#) r!Zstatsmodels.compat.pythonrZnumpyrZstatsmodels.tools.testingrrr#r r r r s