← Back
Editing: _hypotests.cpython-311.pyc
� d�c�2 � � � d dl mZ d dlmZ d dlmZ d dlZd dlZd dl m Z d dl Z d dl mZ ddlmZ dd lmZ dd lmZmZ d d lmZmZmZ d d lmZ dd l m Z ddl m Z m Z! d dl m"Z" g d�Z# edd� � Z$d9d�Z%d dd�d�Z&d� Z' G d� d� � Z(d� Z)d � Z*d:d �Z+d;d �Z,d � Z-d!� Z.d"� Z/d<d#�Z0 ed$d%� � Z1d=d&�Z2d'� Z3d(� Z4 ed)d*e5fd+e5fg� � Z6d>d.�Z7 ed/d*e5fd+e5fg� � Z8d?d0�Z9d1� Z:d2� Z;d@d4�Z< G d5� d6� � Z=d7� Z>d8� Z?dS )A� )� namedtuple)�make_dataclass)�combN)� combinations)�shgo� )� distributions)�ConfidenceInterval)�chi2�norm)�gamma�kv�gammaln)�ifft)�_a_ij_Aij_Dij2)�_concordant_pairs�_discordant_pairs)� _stats_py)�epps_singleton_2samp�cramervonmises�somersd� barnard_exact�boschloo_exact�cramervonmises_2samp� tukey_hsd�poisson_means_test�Epps_Singleton_2sampResult�� statistic�pvalue�g�������?g�������?c �� � t j | � � t j |� � t j |� � }}} | j dk r't d� | j � � � � �|j dk r't d� |j � � � � �t | � � t |� � }}|dk s|dk r#t d� ||� � � � �t j | � � � � � st d� � �t j |� � � � � st d� � �||z }|j dk r't d� |j � � � � �t j |d � � � � � rt d � � �d d l m } |t j | |f� � � � d z }t j |d � � |z }t j t j || z � � t j || z � � f� � j } t j t j ||z � � t j ||z � � f� � j } t j | j d�� � } t j | j d�� � } ||z | z ||z | z z } t j � | � � }t j � |� � }|d t |� � z k rt- j d� � t j | d �� � t j | d �� � z }|t j |j t j ||� � � � z }t5 ||� � dk r dd|dz z d|dz |dz z z z z }||z }t7 j ||� � }t; ||� � S )a� Compute the Epps-Singleton (ES) test statistic. Test the null hypothesis that two samples have the same underlying probability distribution. Parameters ---------- x, y : array-like The two samples of observations to be tested. Input must not have more than one dimension. Samples can have different lengths. t : array-like, optional The points (t1, ..., tn) where the empirical characteristic function is to be evaluated. It should be positive distinct numbers. The default value (0.4, 0.8) is proposed in [1]_. Input must not have more than one dimension. Returns ------- statistic : float The test statistic. pvalue : float The associated p-value based on the asymptotic chi2-distribution. See Also -------- ks_2samp, anderson_ksamp Notes ----- Testing whether two samples are generated by the same underlying distribution is a classical question in statistics. A widely used test is the Kolmogorov-Smirnov (KS) test which relies on the empirical distribution function. Epps and Singleton introduce a test based on the empirical characteristic function in [1]_. One advantage of the ES test compared to the KS test is that is does not assume a continuous distribution. In [1]_, the authors conclude that the test also has a higher power than the KS test in many examples. They recommend the use of the ES test for discrete samples as well as continuous samples with at least 25 observations each, whereas `anderson_ksamp` is recommended for smaller sample sizes in the continuous case. The p-value is computed from the asymptotic distribution of the test statistic which follows a `chi2` distribution. If the sample size of both `x` and `y` is below 25, the small sample correction proposed in [1]_ is applied to the test statistic. The default values of `t` are determined in [1]_ by considering various distributions and finding good values that lead to a high power of the test in general. Table III in [1]_ gives the optimal values for the distributions tested in that study. The values of `t` are scaled by the semi-interquartile range in the implementation, see [1]_. References ---------- .. [1] T. W. Epps and K. J. Singleton, "An omnibus test for the two-sample problem using the empirical characteristic function", Journal of Statistical Computation and Simulation 26, p. 177--203, 1986. .. [2] S. J. Goerg and J. Kaiser, "Nonparametric testing of distributions - the Epps-Singleton two-sample test using the empirical characteristic function", The Stata Journal 9(3), p. 454--465, 2009. r z#x must be 1d, but x.ndim equals {}.z#y must be 1d, but y.ndim equals {}.� zIx and y should have at least 5 elements, but len(x) = {} and len(y) = {}.z$x must not contain nonfinite values.z$y must not contain nonfinite values.z#t must be 1d, but t.ndim equals {}.r z&t must contain positive elements only.)�iqr� )���r T)�biasz�Estimated covariance matrix does not have full rank. This indicates a bad choice of the input t and the test might not be consistent.��axis� � �?g������ܿg333333$@g333333��) �np�asarray�ndim� ValueError�format�len�isfinite�all� less_equal�any� scipy.statsr$ �hstack�reshape�vstack�cos�sin�T�cov�linalg�pinv� matrix_rank�warnings�warn�mean�dot�maxr �sfr )�x�y�t�nx�ny�nr$ �sigma�ts�gx�gy�cov_x�cov_y�est_cov� est_cov_inv�r�g_diff�w�corr�ps �8/usr/lib/python3/dist-packages/scipy/stats/_hypotests.pyr r s� � �D �j��m�m�R�Z��]�]�B�J�q�M�M�!�q�A��v��z�z��>�E�E�a�f�M�M�N�N�N��v��z�z��>�E�E�a�f�M�M�N�N�N� ��V�V�S��V�V��B� �Q���B��F�F�� 1�17���B� � �A� A� A� �;�q�>�>� � � � � A��?�@�@�@� �;�q�>�>� � � � � A��?�@�@�@� �R��A� �v��z�z��>�E�E�a�f�M�M�N�N�N� �}�Q���� � � � � C��A�B�B�B� � � � � � � �C�� �1�a�&�!�!� "� "�Q� &�E� ��A�w� � �%� '�B� ��B�F�2�a�4�L�L�"�&��A��,�,�/� 0� 0� 2�B� ��B�F�2�a�4�L�L�"�&��A��,�,�/� 0� 0� 2�B� �F�2�4�d� #� #� #�E� �F�2�4�d� #� #� #�E���t�U�l�a��d�E�\�)�G��)�.�.��)�)�K� � � � �k�*�*�A��1�S��V�V�8�|�|�� � 6� 7� 7� 7� �W�R�a� � � �2�7�2�A�#6�#6�#6� 6�F� �"�&���2�6�+�v�6�6� 7� 7�7�A� �B�� � �b����C�!�e�*�$�t�R�$�Z�"�t�*�-D�'E�E�F�� �1�H�� ���1� � �A� %�a�� +� +�+� � two-sided)�diff� alternativec �H � t | |||||� � | |z ||z z ||z ||z z z }|dk rt j dd� � S | |dz z ||dz z z }| |z ||z z |z t j |� � z }|||z z } ||z } t j � ddg| � � \ } } t j � ddg| � � \ } }t j | | dz � � }t j | |dz � � dd�df }t j � || � � }t j � || � � }||z }||z }||z |z }||z ||z z }t j dd�� � 5 |t j |� � z }ddd� � n # 1 swxY w Y |d k r+t j |� � t j |� � k }n|d k r||k }n||k }t j ||z | � � }t j ||� � S ) a� Performs the Poisson means test, AKA the "E-test". This is a test of the null hypothesis that the difference between means of two Poisson distributions is `diff`. The samples are provided as the number of events `k1` and `k2` observed within measurement intervals (e.g. of time, space, number of observations) of sizes `n1` and `n2`. Parameters ---------- k1 : int Number of events observed from distribution 1. n1: float Size of sample from distribution 1. k2 : int Number of events observed from distribution 2. n2 : float Size of sample from distribution 2. diff : float, default=0 The hypothesized difference in means between the distributions underlying the samples. alternative : {'two-sided', 'less', 'greater'}, optional Defines the alternative hypothesis. The following options are available (default is 'two-sided'): * 'two-sided': the difference between distribution means is not equal to `diff` * 'less': the difference between distribution means is less than `diff` * 'greater': the difference between distribution means is greater than `diff` Returns ------- statistic : float The test statistic (see [1]_ equation 3.3). pvalue : float The probability of achieving such an extreme value of the test statistic under the null hypothesis. Notes ----- Let: .. math:: X_1 \sim \mbox{Poisson}(\mathtt{n1}\lambda_1) be a random variable independent of .. math:: X_2 \sim \mbox{Poisson}(\mathtt{n2}\lambda_2) and let ``k1`` and ``k2`` be the observed values of :math:`X_1` and :math:`X_2`, respectively. Then `poisson_means_test` uses the number of observed events ``k1`` and ``k2`` from samples of size ``n1`` and ``n2``, respectively, to test the null hypothesis that .. math:: H_0: \lambda_1 - \lambda_2 = \mathtt{diff} A benefit of the E-test is that it has good power for small sample sizes, which can reduce sampling costs [1]_. It has been evaluated and determined to be more powerful than the comparable C-test, sometimes referred to as the Poisson exact test. References ---------- .. [1] Krishnamoorthy, K., & Thomson, J. (2004). A more powerful test for comparing two Poisson means. Journal of Statistical Planning and Inference, 119(1), 23-35. .. [2] Przyborowski, J., & Wilenski, H. (1940). Homogeneity of results in testing samples from Poisson series: With an application to testing clover seed for dodder. Biometrika, 31(3/4), 313-323. Examples -------- Suppose that a gardener wishes to test the number of dodder (weed) seeds in a sack of clover seeds that they buy from a seed company. It has previously been established that the number of dodder seeds in clover follows the Poisson distribution. A 100 gram sample is drawn from the sack before being shipped to the gardener. The sample is analyzed, and it is found to contain no dodder seeds; that is, `k1` is 0. However, upon arrival, the gardener draws another 100 gram sample from the sack. This time, three dodder seeds are found in the sample; that is, `k2` is 3. The gardener would like to know if the difference is significant and not due to chance. The null hypothesis is that the difference between the two samples is merely due to chance, or that :math:`\lambda_1 - \lambda_2 = \mathtt{diff}` where :math:`\mathtt{diff} = 0`. The alternative hypothesis is that the difference is not due to chance, or :math:`\lambda_1 - \lambda_2 \ne 0`. The gardener selects a significance level of 5% to reject the null hypothesis in favor of the alternative [2]_. >>> import scipy.stats as stats >>> res = stats.poisson_means_test(0, 100, 3, 100) >>> res.statistic, res.pvalue (-1.7320508075688772, 0.08837900929018157) The p-value is .088, indicating a near 9% chance of observing a value of the test statistic under the null hypothesis. This exceeds 5%, so the gardener does not reject the null hypothesis as the difference cannot be regarded as significant at this level. r r r% g�����|�=g�������?N�ignore)�invalid�divider\ �less) �_poisson_means_test_ivr �SignificanceResultr, �sqrtr �poisson�ppf�arange�pmf�errstate�abs�sum)�k1�n1�k2�n2r] r^ � lmbd_hat2�var�t_k1k2� nlmbd_hat1� nlmbd_hat2�x1_lb�x1_ub�x2_lb�x2_ub�x1�x2�prob_x1�prob_x2�lmbd_x1�lmbd_x2� lmbds_diff�var_x1x2�t_x1x2� indicatorr s rZ r r � s� � �V �2�r�2�r�4��=�=�=� �r�'�b�2�g�&����b�2�g�)>�>�I� �A�~�~��+�A�q�1�1�1� ��a��.�2��q��>� )�C� �2�g��R�� �$�&�"�'�#�,�,� 6�F� �y�4�'�(�J��i� �J� !�(�,�,�e�Y�-?��L�L�L�E�5� �(�,�,�e�Y�-?��L�L�L�E�5� ��5�%�!�)� $� $�B� ��5�%�!�)� $� $�Q�Q�Q��W� -�B� �#�'�'��J�7�7�G��#�'�'��J�7�7�G� �2�g�G��2�g�G��7�"�T�)�J���|�g��l�*�H� ��X�h� 7� 7� 7� 0� 0��b�g�h� /� /�/��0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0���� 0� 0� 0� 0� �k�!�!��F�6�N�N�b�f�V�n�n�4� � � �� � ��f�$� � ��f�$� � �V�W�w�&� �2� 3� 3�F� � '��� 7� 7�7s �?F#�#F'�*F'c �d � | t | � � k s|t |� � k rt d� � �d}| dk s|dk rt |� � �|dk s|dk rt d� � �|dk rt d� � �h d�}|� � � |vrt d|� d�� � �d S ) Nz `k1` and `k2` must be integers.z1`k1` and `k2` must be greater than or equal to 0.r z%`n1` and `n2` must be greater than 0.z(diff must be greater than or equal to 0.> rc �greaterr\ z Alternative must be one of 'z'.)�int� TypeErrorr/ �lower)rn ro rp rq r] r^ � count_err� alternativess rZ rd rd N s� � � �S��W�W�}�}��c�"�g�g� � ��9�:�:�:�C�I� �A�v�v��a�����#�#�#� �Q�w�w�"��'�'��@�A�A�A� �a�x�x��C�D�D�D�3�3�3�L������,�.�.��H� �H�H�H�I�I�I� /�.r[ c � � e Zd Zd� Zd� ZdS )�CramerVonMisesResultc �" � || _ || _ d S �Nr )�selfr r s rZ �__init__z CramerVonMisesResult.__init__c s � �"��� �� � � r[ c �@ � | j j � d| j � d| j � d�S )Nz (statistic=z , pvalue=�))� __class__�__name__r r )r� s rZ �__repr__z CramerVonMisesResult.__repr__g s; � ��>�*� )� )�t�~� )� )� �+�)� )� )� *r[ N)r� � __module__� __qualname__r� r� � r[ rZ r� r� b s2 � � � � � � � � �*� *� *� *� *r[ r� c � ��� d� �d� ���fd�}t j | � � } t j | d�� � }t j | d�� � }d}t j |� � ru ||| | � � t j t |dz � � z z }|| |z ||< t j |� � d k ||< |dz }t j |� � �u|S ) u psi1 is defined in equation 1.10 in Csörgő, S. and Faraway, J. (1996). This implements a modified version by excluding the term V(x) / 12 (here: _cdf_cvm_inf(x) / 12) to avoid evaluating _cdf_cvm_inf(x) twice in _cdf_cvm. Implementation based on MAPLE code of Julian Faraway and R code of the function pCvM in the package goftest (v1.1.1), permission granted by Adrian Baddeley. Main difference in the implementation: the code here keeps adding terms of the series until the terms are small enough. c �� � | dz dz }t d|� � t d|� � z }t j | � � | dz dz z |z t j t j � � z S )Nr% � � �?� �?� �?)r r, �exprf �pi)rH �z�bs rZ �_ed2z_psi1_mod.<locals>._ed2y sZ � � �q�D�1�H�� �s�A�J�J��C���� #���v�q�b�z�z�Q�q�S�C�L�(�1�,�r�w�r�u�~�~�=�=r[ c � � | dz dz }t j | � � t j t j � � z }|| dz dz z dt d|� � z dt d|� � z z t d|� � z z S )Nr% r� g @r� � r� � �?)r, r� rf r� r )rH r� �cs rZ �_ed3z_psi1_mod.<locals>._ed3~ so � � �q�D�1�H�� �F�A�2�J�J����� � � '���A�a�C�3�<� �1�R��Q�Z�Z�<�!�B�s�A�J�J�,�#>��C����#K�L�Lr[ c �d �� d| z dz }dt j |� � z }|dz }|dz }|t | dz � � z � d| z dz |z � � z d|z z }t | dz � � � d| z dz |z � � z d |z z }d|dz z t | d z � � z � d| z d z |z � � z d |z z }d |z t | dz � � z � d| z dz |z � � z d|z z } d |z t | dz � � z � d| z d z |z � � z d|z z } ||z |z | z | z S )Nr% r r� r� � �?r� r� � �H r� r# � � � )r, rf r ) �krG �m�sx�y1�y2�e1�e2�e3�e4�e5r� r� s ��rZ �_Akz_psi1_mod.<locals>._Ak� sr �� � �a�C�!�G�� ������^�� ��X�� ��X�� ��q�3�w� � � �$�$��A��� �2�~�"6�"6� 6�!�b�&� A�� �1�s�7�^�^�d�d�A��E�A�I��#3� 4� 4� 4��R�� @�� �!�a�%�[�5��S��>�>� )�D�D�!�a�%�!�)�r�1A�,B�,B� B�b�2�g� N�� ��U�U�1�s�7�^�^� #�d�d�A��E�A�I��+;�&<�&<� <��b�� I�� ��U�U�1�s�7�^�^� #�d�d�A��E�A�I��+;�&<�&<� <��b�� I���B�w��|�b� �2�%�%r[ �float��dtype�boolr r �H�����z>)r, r- � zeros_like� ones_liker5 r� r rl )rG r� �tot�condr� r� r� r� s @@rZ � _psi1_modr� l s ��� �>� >� >� M� M� M� &� &� &� &� &� &� � �1� � �A� �-��� )� )� )�C� �<��� (� (� (�D� �A� �&��,�,� � �S��A�d�G�_�_� ����a�!�e� � � 4� 5����I��M��D� ��V�A�Y�Y�$�&��T� � �Q��� �&��,�,� � �Jr[ c �` � t j | � � } d� }t j | d�� � }t j | d�� � }d}t j |� � rT || | |� � }|| |z ||< t j |� � dk ||< |dz }t j |� � �T|S )u2 Calculate the cdf of the Cramér-von Mises statistic (infinite sample size). See equation 1.2 in Csörgő, S. and Faraway, J. (1996). Implementation based on MAPLE code of Julian Faraway and R code of the function pCvM in the package goftest (v1.1.1), permission granted by Adrian Baddeley. Main difference in the implementation: the code here keeps adding terms of the series until the terms are small enough. The function is not expected to be accurate for large values of x, say x > 4, when the cdf is very close to 1. c �` � t j t |dz � � t |dz � � z � � t j dz t j | � � z z }d|z dz }|dz d| z z }t d|� � }|t j |� � z t j | � � z |z S )Nr� r r� r� r% � r� )r, r� r r� rf r )rG r� �urH �qr� s rZ �termz_cdf_cvm_inf.<locals>.term� s� � � �F�7�1�s�7�#�#�g�a��c�l�l�2� 3� 3�r�u�c�z�B�G�A�J�J�7N� O�� �a�C�!�G�� �q�D�B�q�D�M�� �t�Q�K�K���2�7�1�:�:�~����r� � �*�Q�.�.r[ r� r� r� r r� r )r, r- r� r� r5 rl )rG r� r� r� r� r� s rZ � _cdf_cvm_infr� � s� � � � �1� � �A�/� /� /� �-��� )� )� )�C� �<��� (� (� (�D� �A� �&��,�,� � �D��4��!� � ����I��M��D� ��V�A�Y�Y�$�&��T� � �Q��� �&��,�,� � �Jr[ c �b � t j | � � } |�t | � � }nut j | d�� � }dd|z z | k | |dz k z }t | | � � ddd|z z z z t | | � � |z z ||< d|| |dz k < |j d k r|d S |S ) u� Calculate the cdf of the Cramér-von Mises statistic for a finite sample size n. If N is None, use the asymptotic cdf (n=inf). See equation 1.8 in Csörgő, S. and Faraway, J. (1996) for finite samples, 1.2 for the asymptotic cdf. The function is not expected to be accurate for large values of x, say x > 2, when the cdf is very close to 1 and it might return values > 1 in that case, e.g. _cdf_cvm(2.0, 12) = 1.0000027556716846. Moreover, it is not accurate for small values of n, especially close to the bounds of the distribution's domain, [1/(12*n), n/3], where the value jumps to 0 and 1, respectively. These are limitations of the approximation by Csörgő and Faraway (1996) implemented in this function. Nr� r� r+ r� g @r r� r r� )r, r- r� r� r� r. )rG rL rH �sups rZ �_cdf_cvmr� � s� � � � �1� � �A��y� ��O�O��� �M�!�7� +� +� +���2�a�4�y�1�}��Q�r�T��*�� �a��f�%�%��R��A��Y� �7�)�A�c�F�:K�:K�a�:O�O��#����!�q��s�(� ��v��{�{���u� � �Hr[ r� c �6 � t |t � � rt t |� � j }t j t j | � � � � }|j dk rt d� � �|j dk rt d� � �t |� � } ||g|�R � }dt j d|dz � � z dz d|z z }dd|z z t j ||z dz � � z }t ddt ||� � z � � }t! ||�� � S ) uf Perform the one-sample Cramér-von Mises test for goodness of fit. This performs a test of the goodness of fit of a cumulative distribution function (cdf) :math:`F` compared to the empirical distribution function :math:`F_n` of observed random variates :math:`X_1, ..., X_n` that are assumed to be independent and identically distributed ([1]_). The null hypothesis is that the :math:`X_i` have cumulative distribution :math:`F`. Parameters ---------- rvs : array_like A 1-D array of observed values of the random variables :math:`X_i`. cdf : str or callable The cumulative distribution function :math:`F` to test the observations against. If a string, it should be the name of a distribution in `scipy.stats`. If a callable, that callable is used to calculate the cdf: ``cdf(x, *args) -> float``. args : tuple, optional Distribution parameters. These are assumed to be known; see Notes. Returns ------- res : object with attributes statistic : float Cramér-von Mises statistic. pvalue : float The p-value. See Also -------- kstest, cramervonmises_2samp Notes ----- .. versionadded:: 1.6.0 The p-value relies on the approximation given by equation 1.8 in [2]_. It is important to keep in mind that the p-value is only accurate if one tests a simple hypothesis, i.e. the parameters of the reference distribution are known. If the parameters are estimated from the data (composite hypothesis), the computed p-value is not reliable. References ---------- .. [1] Cramér-von Mises criterion, Wikipedia, https://en.wikipedia.org/wiki/Cram%C3%A9r%E2%80%93von_Mises_criterion .. [2] Csörgő, S. and Faraway, J. (1996). The Exact and Asymptotic Distribution of Cramér-von Mises Statistics. Journal of the Royal Statistical Society, pp. 221-234. Examples -------- Suppose we wish to test whether data generated by ``scipy.stats.norm.rvs`` were, in fact, drawn from the standard normal distribution. We choose a significance level of alpha=0.05. >>> import numpy as np >>> from scipy import stats >>> rng = np.random.default_rng() >>> x = stats.norm.rvs(size=500, random_state=rng) >>> res = stats.cramervonmises(x, 'norm') >>> res.statistic, res.pvalue (0.49121480855028343, 0.04189256516661377) The p-value 0.79 exceeds our chosen significance level, so we do not reject the null hypothesis that the observed sample is drawn from the standard normal distribution. Now suppose we wish to check whether the same samples shifted by 2.1 is consistent with being drawn from a normal distribution with a mean of 2. >>> y = x + 2.1 >>> res = stats.cramervonmises(y, 'norm', args=(2,)) >>> res.statistic, res.pvalue (0.07400330012187435, 0.7274595666160468) Here we have used the `args` keyword to specify the mean (``loc``) of the normal distribution to test the data against. This is equivalent to the following, in which we create a frozen normal distribution with mean 2.1, then pass its ``cdf`` method as an argument. >>> frozen_dist = stats.norm(loc=2) >>> res = stats.cramervonmises(y, frozen_dist.cdf) >>> res.statistic, res.pvalue (0.07400330012187435, 0.7274595666160468) In either case, we would reject the null hypothesis that the observed sample is drawn from a normal distribution with a mean of 2 (and default variance of 1) because the p-value 0.04 is less than our chosen significance level. r z2The sample must contain at least two observations.z#The sample must be one-dimensional.r% r� r r+ r )� isinstance�str�getattrr �cdfr, �sortr- �sizer/ r. r1 ri rm rE r� r� ) �rvsr� �args�valsrL �cdfvalsr� rW rY s rZ r r � s � �~ �#�s��� .��m�S�)�)�-�� �7�2�:�c�?�?� #� #�D� �y�A�~�~��M�N�N�N� �y�1�}�}��>�?�?�?� �D� � �A��c�$� �� � � �G� �2�9�Q��!�� � � �q� �1�Q�3�'�A� �2�a�4��2�6�1�w�;��*�+�+�+�A� �A�r�H�Q��N�N�"�#�#�A� �!�A� 6� 6� 6�6r[ c �, � t j dt j �� � }t d| dz � � D ]_}|}t j ||dz z dz dz t j �� � }t |� � }|dz |d|�< || d�xx |dz z cc< �`|S )z� Distribution of probability of the Wilcoxon ranksum statistic r_plus (sum of ranks of positive differences). Returns an array with the probabilities of all the possible ranks r = 0, ..., n*(n+1)/2 r r� r% r� N)r, �ones�double�range�zerosr1 )rL r� r� �prev_cr� s rZ �_get_wilcoxon_distrr� X s� � � �����#�#�#�A� �1�a�!�e�_�_� � ���� �H�Q�!�a�%�[�A�%��)��� ;� ;� ;�� ��K�K���� ��"�1�"�� �1�"�#�#����&�3�,� ����� �Hr[ c �, � t j d| dz � � dd�df }| | dz z dz }d|z }t j |� � }dt j z |z |z }t j t j ||z � � d�� � }t j d|z |z � � |z }t j t |� � � � }t j t |� � dz � � } |ddd� | dd�< | dxx dz cc< | d | d< | S )a� Distribution of probability of the Wilcoxon ranksum statistic r_plus (sum of ranks of positive differences). Returns an array with the probabilities of all the possible ranks r = 0, ..., n*(n+1)/2 This is a slower reference function References ---------- .. [1] 1. Harris T, Hardin JW. Exact Wilcoxon Signed-Rank and Wilcoxon Mann-Whitney Ranksum Tests. The Stata Journal. 2013;13(2):337-343. r Nr% r r( y �?r& ) r, ri r� �prodr: r� �realr r� r� ) rL �airI r� �j�theta�phi_sp�phi_srY �ress rZ �_get_wilcoxon_distr2r� i s � � ��1�a��c� � �1�1�1�d�7� #�B� �1�Q�3��� �A� �!��A� � �!� � �A� �b�e�G�A�I�a�K�E� �W�R�V�E�"�H�%�%�A� .� .� .�F� �F�2�e�8�A�:� � �� '�E� ���U� � � � �A� �(�3�q�6�6�!�8� � �C��#�#�A�#��C����I���F�F�F�a�K�F�F�F��!�f�C��G� �Jr[ c �x � | j d dk s| j d dk rt j t j fS | � � � }t | � � }t | � � }| � d�� � dz � � � }| � d�� � dz � � � }|dz |z |dz |z z }||z |dz z }dt | � � ||z dz |z z z }||z } | dk r|dfS || dz z } dt j t | � � � � z } || fS )z=Calculate Kendall's tau-b and p-value from contingency table.r r r( r% r� r� ) �shaper, �nanrm �_P�_Qr r rF rl ) �A�NA�PA�QA�Sri2�Scj2� denominator�tau� numerator� s02_tau_b�ZrY s rZ �_tau_br� � s2 � � �w�q�z�Q���!�'�!�*��/�/��v�r�v�~� � �����B� �A���B� �A���B� �E�E�q�E�M�M�1� � !� !� #� #�D� �E�E�q�E�M�M�1� � !� !� #� #�D��q�5�4�<�"�a�%�$�,�/�K� �b�5�;��$� $�C��>�!�$�$��R��!�|�b�'8�8�9�I��+�%�I��A�~�~��A�v� � �I�s�N��A� �$�'�#�a�&�&�/�/��A� ��6�Mr[ c �T � | j d dk s| j d dk rt j t j fS | � � � }|dz }t | � � }t | � � }| � d�� � dz � � � }||z ||z z }t | � � ||z dz |z z }t j d�� � 5 ||z d|z dz z } d d d � � n # 1 swxY w Y t j j � | |� � \ } } || fS ) z7Calculate Somers' D and p-value from contingency table.r r r% r( r` )rb r� r� N) r� r, r� rm r� r� r rk �scipy�statsr �_normtest_finish) r� r^ r� �NA2r� r� r� �d�Sr� �_rY s rZ � _somers_dr � sT � � �w�q�z�Q���!�'�!�*��/�/��v�r�v�~� � �����B� �a�%�C� �A���B� �A���B� �E�E�q�E�M�M�1� � !� !� #� #�D� �b��3��:� �A��q���R��U�Q�J�r�M�)�A� ��H� %� %� %� #� #� �"�W�q�!�u�s�l� "��#� #� #� #� #� #� #� #� #� #� #���� #� #� #� #� �;� � 1� 1�!�[� A� A�D�A�q� �a�4�Ks �C3�3C7�:C7� SomersDResult)r r �tablec �� � t j | � � t j |� � }} | j dk rK| j |j k rt d� � �t j j � | |� � d }n�| j dk r�t j | dk � � rt d� � �t j | | � t � � k � � rt d� � �| � � � d j dk rt d� � �| }nt d� � �t ||� � \ }}t |||� � }||_ |S ) a Calculates Somers' D, an asymmetric measure of ordinal association. Like Kendall's :math:`\tau`, Somers' :math:`D` is a measure of the correspondence between two rankings. Both statistics consider the difference between the number of concordant and discordant pairs in two rankings :math:`X` and :math:`Y`, and both are normalized such that values close to 1 indicate strong agreement and values close to -1 indicate strong disagreement. They differ in how they are normalized. To show the relationship, Somers' :math:`D` can be defined in terms of Kendall's :math:`\tau_a`: .. math:: D(Y|X) = \frac{\tau_a(X, Y)}{\tau_a(X, X)} Suppose the first ranking :math:`X` has :math:`r` distinct ranks and the second ranking :math:`Y` has :math:`s` distinct ranks. These two lists of :math:`n` rankings can also be viewed as an :math:`r \times s` contingency table in which element :math:`i, j` is the number of rank pairs with rank :math:`i` in ranking :math:`X` and rank :math:`j` in ranking :math:`Y`. Accordingly, `somersd` also allows the input data to be supplied as a single, 2D contingency table instead of as two separate, 1D rankings. Note that the definition of Somers' :math:`D` is asymmetric: in general, :math:`D(Y|X) \neq D(X|Y)`. ``somersd(x, y)`` calculates Somers' :math:`D(Y|X)`: the "row" variable :math:`X` is treated as an independent variable, and the "column" variable :math:`Y` is dependent. For Somers' :math:`D(X|Y)`, swap the input lists or transpose the input table. Parameters ---------- x : array_like 1D array of rankings, treated as the (row) independent variable. Alternatively, a 2D contingency table. y : array_like, optional If `x` is a 1D array of rankings, `y` is a 1D array of rankings of the same length, treated as the (column) dependent variable. If `x` is 2D, `y` is ignored. alternative : {'two-sided', 'less', 'greater'}, optional Defines the alternative hypothesis. Default is 'two-sided'. The following options are available: * 'two-sided': the rank correlation is nonzero * 'less': the rank correlation is negative (less than zero) * 'greater': the rank correlation is positive (greater than zero) Returns ------- res : SomersDResult A `SomersDResult` object with the following fields: statistic : float The Somers' :math:`D` statistic. pvalue : float The p-value for a hypothesis test whose null hypothesis is an absence of association, :math:`D=0`. See notes for more information. table : 2D array The contingency table formed from rankings `x` and `y` (or the provided contingency table, if `x` is a 2D array) See Also -------- kendalltau : Calculates Kendall's tau, another correlation measure. weightedtau : Computes a weighted version of Kendall's tau. spearmanr : Calculates a Spearman rank-order correlation coefficient. pearsonr : Calculates a Pearson correlation coefficient. Notes ----- This function follows the contingency table approach of [2]_ and [3]_. *p*-values are computed based on an asymptotic approximation of the test statistic distribution under the null hypothesis :math:`D=0`. Theoretically, hypothesis tests based on Kendall's :math:`tau` and Somers' :math:`D` should be identical. However, the *p*-values returned by `kendalltau` are based on the null hypothesis of *independence* between :math:`X` and :math:`Y` (i.e. the population from which pairs in :math:`X` and :math:`Y` are sampled contains equal numbers of all possible pairs), which is more specific than the null hypothesis :math:`D=0` used here. If the null hypothesis of independence is desired, it is acceptable to use the *p*-value returned by `kendalltau` with the statistic returned by `somersd` and vice versa. For more information, see [2]_. Contingency tables are formatted according to the convention used by SAS and R: the first ranking supplied (``x``) is the "row" variable, and the second ranking supplied (``y``) is the "column" variable. This is opposite the convention of Somers' original paper [1]_. References ---------- .. [1] Robert H. Somers, "A New Asymmetric Measure of Association for Ordinal Variables", *American Sociological Review*, Vol. 27, No. 6, pp. 799--811, 1962. .. [2] Morton B. Brown and Jacqueline K. Benedetti, "Sampling Behavior of Tests for Correlation in Two-Way Contingency Tables", *Journal of the American Statistical Association* Vol. 72, No. 358, pp. 309--315, 1977. .. [3] SAS Institute, Inc., "The FREQ Procedure (Book Excerpt)", *SAS/STAT 9.2 User's Guide, Second Edition*, SAS Publishing, 2009. .. [4] Laerd Statistics, "Somers' d using SPSS Statistics", *SPSS Statistics Tutorials and Statistical Guides*, https://statistics.laerd.com/spss-tutorials/somers-d-using-spss-statistics.php, Accessed July 31, 2020. Examples -------- We calculate Somers' D for the example given in [4]_, in which a hotel chain owner seeks to determine the association between hotel room cleanliness and customer satisfaction. The independent variable, hotel room cleanliness, is ranked on an ordinal scale: "below average (1)", "average (2)", or "above average (3)". The dependent variable, customer satisfaction, is ranked on a second scale: "very dissatisfied (1)", "moderately dissatisfied (2)", "neither dissatisfied nor satisfied (3)", "moderately satisfied (4)", or "very satisfied (5)". 189 customers respond to the survey, and the results are cast into a contingency table with the hotel room cleanliness as the "row" variable and customer satisfaction as the "column" variable. +-----+-----+-----+-----+-----+-----+ | | (1) | (2) | (3) | (4) | (5) | +=====+=====+=====+=====+=====+=====+ | (1) | 27 | 25 | 14 | 7 | 0 | +-----+-----+-----+-----+-----+-----+ | (2) | 7 | 14 | 18 | 35 | 12 | +-----+-----+-----+-----+-----+-----+ | (3) | 1 | 3 | 2 | 7 | 17 | +-----+-----+-----+-----+-----+-----+ For example, 27 customers assigned their room a cleanliness ranking of "below average (1)" and a corresponding satisfaction of "very dissatisfied (1)". We perform the analysis as follows. >>> from scipy.stats import somersd >>> table = [[27, 25, 14, 7, 0], [7, 14, 18, 35, 12], [1, 3, 2, 7, 17]] >>> res = somersd(table) >>> res.statistic 0.6032766111513396 >>> res.pvalue 1.0007091191074533e-27 The value of the Somers' D statistic is approximately 0.6, indicating a positive correlation between room cleanliness and customer satisfaction in the sample. The *p*-value is very small, indicating a very small probability of observing such an extreme value of the statistic under the null hypothesis that the statistic of the entire population (from which our sample of 189 customers is drawn) is zero. This supports the alternative hypothesis that the true value of Somers' D for the population is nonzero. r z!Rankings must be of equal length.r% r z;All elements of the contingency table must be non-negative.z6All elements of the contingency table must be integer.z?At least two elements of the contingency table must be nonzero.z!x must be either a 1D or 2D array)r, �arrayr. r� r/ r� r� � contingency�crosstabr5 �astyper� �nonzeror r � correlation)rG rH r^ r r rY r� s rZ r r � sO � �v �8�A�;�;���� � �q�A��v��{�{� �6�Q�V� � � � @�A�A� A�� �'�0�0��A�6�6�q�9��� ��1��� �6�!�a�%�=�=� .� � -� .� .� .� �6�!�q�x�x��}�}�$� %� %� )� � (� )� )� )� �9�9�;�;�q�>� �� "� "� � 0� 1� 1� 1�����<�=�=�=� �U�K� (� (�D�A�q� ��1�e� $� $�C��C�O� �Jr[ c # �� K � t j | |z � � }t || � � D ]F}t j |� � }t j | |z t � � }d||< || }||fV � �GdS )z� Partition a set of indices into two fixed-length sets in all possible ways Partition a set of indices 0 ... nx + ny - 1 into two sets of length nx and ny in all possible ways (ignoring order of elements). FN)r, ri r r r� r� )rJ rK r� r� rG �maskrH s rZ �_all_partitionsr s s� � � � � � �"�R�%���A� �!�R� � � � �� �H�Q�K�K���w�r�"�u�d�#�#����Q�� �d�G����d� � � � � � r[ c � � t t j | dz � � dz � � }t | dz � � |z |ddd� z S )z'Compute all log combination of C(n, k).r Nr& )r r, ri )rL � gammaln_arrs rZ �_compute_log_combinationsr � sG � ��"�)�A��E�*�*�Q�.�/�/�K� �1�q�5�>�>�K� '�+�d�d��d�*;� ;�;r[ �BarnardExactResultr r T� c � � |dk rt d|��� � �t j | t j �� � } | j dk st d� � �t j | dk � � rt d� � �d| � d�� � v rt t j d� � S | � d�� � \ }}t j |d z t j �� � � d d � � }t j |d z t j �� � � d d � � }||z ||z } }|r ||z ||z z } | d | z z d |z d |z z z } n|d |z z |z | d | z z |z z } t j d d � � � 5 t j || z t j | � � � � } d d d � � n # 1 swxY w Y d| || k < | | d | d f } |dk r&t j | � � t | � � k }n.|dk r| | k }n!|dk r| | k }nd|��}t |� � �||z }t |� � }t |� � }|| || z }t! t" |||fd|d�� � }t j t j |j � � dd �� � }t | |� � S )a� Perform a Barnard exact test on a 2x2 contingency table. Parameters ---------- table : array_like of ints A 2x2 contingency table. Elements should be non-negative integers. alternative : {'two-sided', 'less', 'greater'}, optional Defines the null and alternative hypotheses. Default is 'two-sided'. Please see explanations in the Notes section below. pooled : bool, optional Whether to compute score statistic with pooled variance (as in Student's t-test, for example) or unpooled variance (as in Welch's t-test). Default is ``True``. n : int, optional Number of sampling points used in the construction of the sampling method. Note that this argument will automatically be converted to the next higher power of 2 since `scipy.stats.qmc.Sobol` is used to select sample points. Default is 32. Must be positive. In most cases, 32 points is enough to reach good precision. More points comes at performance cost. Returns ------- ber : BarnardExactResult A result object with the following attributes. statistic : float The Wald statistic with pooled or unpooled variance, depending on the user choice of `pooled`. pvalue : float P-value, the probability of obtaining a distribution at least as extreme as the one that was actually observed, assuming that the null hypothesis is true. See Also -------- chi2_contingency : Chi-square test of independence of variables in a contingency table. fisher_exact : Fisher exact test on a 2x2 contingency table. boschloo_exact : Boschloo's exact test on a 2x2 contingency table, which is an uniformly more powerful alternative to Fisher's exact test. Notes ----- Barnard's test is an exact test used in the analysis of contingency tables. It examines the association of two categorical variables, and is a more powerful alternative than Fisher's exact test for 2x2 contingency tables. Let's define :math:`X_0` a 2x2 matrix representing the observed sample, where each column stores the binomial experiment, as in the example below. Let's also define :math:`p_1, p_2` the theoretical binomial probabilities for :math:`x_{11}` and :math:`x_{12}`. When using Barnard exact test, we can assert three different null hypotheses : - :math:`H_0 : p_1 \geq p_2` versus :math:`H_1 : p_1 < p_2`, with `alternative` = "less" - :math:`H_0 : p_1 \leq p_2` versus :math:`H_1 : p_1 > p_2`, with `alternative` = "greater" - :math:`H_0 : p_1 = p_2` versus :math:`H_1 : p_1 \neq p_2`, with `alternative` = "two-sided" (default one) In order to compute Barnard's exact test, we are using the Wald statistic [3]_ with pooled or unpooled variance. Under the default assumption that both variances are equal (``pooled = True``), the statistic is computed as: .. math:: T(X) = \frac{ \hat{p}_1 - \hat{p}_2 }{ \sqrt{ \hat{p}(1 - \hat{p}) (\frac{1}{c_1} + \frac{1}{c_2}) } } with :math:`\hat{p}_1, \hat{p}_2` and :math:`\hat{p}` the estimator of :math:`p_1, p_2` and :math:`p`, the latter being the combined probability, given the assumption that :math:`p_1 = p_2`. If this assumption is invalid (``pooled = False``), the statistic is: .. math:: T(X) = \frac{ \hat{p}_1 - \hat{p}_2 }{ \sqrt{ \frac{\hat{p}_1 (1 - \hat{p}_1)}{c_1} + \frac{\hat{p}_2 (1 - \hat{p}_2)}{c_2} } } The p-value is then computed as: .. math:: \sum \binom{c_1}{x_{11}} \binom{c_2}{x_{12}} \pi^{x_{11} + x_{12}} (1 - \pi)^{t - x_{11} - x_{12}} where the sum is over all 2x2 contingency tables :math:`X` such that: * :math:`T(X) \leq T(X_0)` when `alternative` = "less", * :math:`T(X) \geq T(X_0)` when `alternative` = "greater", or * :math:`T(X) \geq |T(X_0)|` when `alternative` = "two-sided". Above, :math:`c_1, c_2` are the sum of the columns 1 and 2, and :math:`t` the total (sum of the 4 sample's element). The returned p-value is the maximum p-value taken over the nuisance parameter :math:`\pi`, where :math:`0 \leq \pi \leq 1`. This function's complexity is :math:`O(n c_1 c_2)`, where `n` is the number of sample points. References ---------- .. [1] Barnard, G. A. "Significance Tests for 2x2 Tables". *Biometrika*. 34.1/2 (1947): 123-138. :doi:`dpgkg3` .. [2] Mehta, Cyrus R., and Pralay Senchaudhuri. "Conditional versus unconditional exact tests for comparing two binomials." *Cytel Software Corporation* 675 (2003): 1-5. .. [3] "Wald Test". *Wikipedia*. https://en.wikipedia.org/wiki/Wald_test Examples -------- An example use of Barnard's test is presented in [2]_. Consider the following example of a vaccine efficacy study (Chan, 1998). In a randomized clinical trial of 30 subjects, 15 were inoculated with a recombinant DNA influenza vaccine and the 15 were inoculated with a placebo. Twelve of the 15 subjects in the placebo group (80%) eventually became infected with influenza whereas for the vaccine group, only 7 of the 15 subjects (47%) became infected. The data are tabulated as a 2 x 2 table:: Vaccine Placebo Yes 7 12 No 8 3 When working with statistical hypothesis testing, we usually use a threshold probability or significance level upon which we decide to reject the null hypothesis :math:`H_0`. Suppose we choose the common significance level of 5%. Our alternative hypothesis is that the vaccine will lower the chance of becoming infected with the virus; that is, the probability :math:`p_1` of catching the virus with the vaccine will be *less than* the probability :math:`p_2` of catching the virus without the vaccine. Therefore, we call `barnard_exact` with the ``alternative="less"`` option: >>> import scipy.stats as stats >>> res = stats.barnard_exact([[7, 12], [8, 3]], alternative="less") >>> res.statistic -1.894... >>> res.pvalue 0.03407... Under the null hypothesis that the vaccine will not lower the chance of becoming infected, the probability of obtaining test results at least as extreme as the observed data is approximately 3.4%. Since this p-value is less than our chosen significance level, we have evidence to reject :math:`H_0` in favor of the alternative. Suppose we had used Fisher's exact test instead: >>> _, pvalue = stats.fisher_exact([[7, 12], [8, 3]], alternative="less") >>> pvalue 0.0640... With the same threshold significance of 5%, we would not have been able to reject the null hypothesis in favor of the alternative. As stated in [2]_, Barnard's test is uniformly more powerful than Fisher's exact test because Barnard's test does not condition on any margin. Fisher's test should only be used when both sets of marginals are fixed. r �6Number of points `n` must be strictly positive, found r� �r% r% �*The input `table` must be of shape (2, 2).�*All values in `table` must be nonnegative.r( r+ r r& r` �rb ra N�r r �r r r\ rc r� zG`alternative` should be one of {'two-sided', 'less', 'greater'}, found �r �sobol�r� �boundsrL �sampling_method��a_min�a_max)r/ r, r- �int64r� r5 rm r r� ri r8 rk rb rf rl r r �-_get_binomial_log_p_value_with_nuisance_param�clipr� �fun)r r^ �pooledrL � total_col_1� total_col_2r{ r| �p1�p2rY � variances�wald_statistic� wald_stat_obs� index_arr�msg� x1_sum_x2� x1_log_comb� x2_log_comb�x1_sum_x2_log_comb�result�p_values rZ r r � sq � �| �A�v�v�� �� � � � � � �J�u�B�H� -� -� -�E� �;�&� � ��E�F�F�F� �v�e�a�i��� G��E�F�F�F��E�I�I�1�I� � � � � "�"�&�#�.�.�.� $�y�y�a�y� 0� 0� �K�� ��;��?�"�(� 3� 3� 3� ;� ;�B�� B� B�B� ��;��?�"�(� 3� 3� 3� ;� ;�A�r� B� B�B� �+� �r�K� /��B� � N� �"�W��{�2� 3����Q��K�1�{�?�Q��_�#D�E� � ��!�b�&�M�K�/�"��B��-�+�2M�M� � ��H�h� 7� 7� 7� B� B���B��G�b�g�i�.@�.@�A�A��B� B� B� B� B� B� B� B� B� B� B���� B� B� B� B� !�N�2��8� �"�5��;��d� �#;�<�M��k�!�!��F�>�*�*�c�-�.@�.@�@� � � �� � �"�m�3� � � � � !� !�"�m�3� � � &�!� &� &� � ��o�o� ��R��I�+�K�8�8�K�+�K�8�8�K�$�R� �;�r�?�:�� �5��+�Y� 7�� � � � � �F� �g�b�f�f�j�[�)�)��!�<�<�<�G� �m�W� 5� 5�5s � +G�G� G�BoschlooExactResultc �� � t j }|dk rt d|��� � �t j | t j �� � } | j dk st d� � �t j | dk � � rt d� � �d| � d�� � v r$t t j t j � � S | � d�� � \ }}||z }t j |dz t j �� � � dd � � }t j |dz t j �� � � d d� � }||z } |d k r |� ||| |� � j } n�|d k r |� ||| |� � j } n�|d k rlt | d |� � � } t | d |� � � } | j | j k r| n| } t j d| j z dd�� � }t | j |� � S dd� d|��}t |� � �| | d | d f }| |dz k }|j |j | j } }}t' |� � }t' |� � }|| || z }t) t* | ||fd|d�� � }t j t j |j � � dd�� � }t ||� � S )a Perform Boschloo's exact test on a 2x2 contingency table. Parameters ---------- table : array_like of ints A 2x2 contingency table. Elements should be non-negative integers. alternative : {'two-sided', 'less', 'greater'}, optional Defines the null and alternative hypotheses. Default is 'two-sided'. Please see explanations in the Notes section below. n : int, optional Number of sampling points used in the construction of the sampling method. Note that this argument will automatically be converted to the next higher power of 2 since `scipy.stats.qmc.Sobol` is used to select sample points. Default is 32. Must be positive. In most cases, 32 points is enough to reach good precision. More points comes at performance cost. Returns ------- ber : BoschlooExactResult A result object with the following attributes. statistic : float The statistic used in Boschloo's test; that is, the p-value from Fisher's exact test. pvalue : float P-value, the probability of obtaining a distribution at least as extreme as the one that was actually observed, assuming that the null hypothesis is true. See Also -------- chi2_contingency : Chi-square test of independence of variables in a contingency table. fisher_exact : Fisher exact test on a 2x2 contingency table. barnard_exact : Barnard's exact test, which is a more powerful alternative than Fisher's exact test for 2x2 contingency tables. Notes ----- Boschloo's test is an exact test used in the analysis of contingency tables. It examines the association of two categorical variables, and is a uniformly more powerful alternative to Fisher's exact test for 2x2 contingency tables. Boschloo's exact test uses the p-value of Fisher's exact test as a statistic, and Boschloo's p-value is the probability under the null hypothesis of observing such an extreme value of this statistic. Let's define :math:`X_0` a 2x2 matrix representing the observed sample, where each column stores the binomial experiment, as in the example below. Let's also define :math:`p_1, p_2` the theoretical binomial probabilities for :math:`x_{11}` and :math:`x_{12}`. When using Boschloo exact test, we can assert three different alternative hypotheses: - :math:`H_0 : p_1=p_2` versus :math:`H_1 : p_1 < p_2`, with `alternative` = "less" - :math:`H_0 : p_1=p_2` versus :math:`H_1 : p_1 > p_2`, with `alternative` = "greater" - :math:`H_0 : p_1=p_2` versus :math:`H_1 : p_1 \neq p_2`, with `alternative` = "two-sided" (default) There are multiple conventions for computing a two-sided p-value when the null distribution is asymmetric. Here, we apply the convention that the p-value of a two-sided test is twice the minimum of the p-values of the one-sided tests (clipped to 1.0). Note that `fisher_exact` follows a different convention, so for a given `table`, the statistic reported by `boschloo_exact` may differ from the p-value reported by `fisher_exact` when ``alternative='two-sided'``. .. versionadded:: 1.7.0 References ---------- .. [1] R.D. Boschloo. "Raised conditional level of significance for the 2 x 2-table when testing the equality of two probabilities", Statistica Neerlandica, 24(1), 1970 .. [2] "Boschloo's test", Wikipedia, https://en.wikipedia.org/wiki/Boschloo%27s_test .. [3] Lise M. Saari et al. "Employee attitudes and job satisfaction", Human Resource Management, 43(4), 395-407, 2004, :doi:`10.1002/hrm.20032`. Examples -------- In the following example, we consider the article "Employee attitudes and job satisfaction" [3]_ which reports the results of a survey from 63 scientists and 117 college professors. Of the 63 scientists, 31 said they were very satisfied with their jobs, whereas 74 of the college professors were very satisfied with their work. Is this significant evidence that college professors are happier with their work than scientists? The following table summarizes the data mentioned above:: college professors scientists Very Satisfied 74 31 Dissatisfied 43 32 When working with statistical hypothesis testing, we usually use a threshold probability or significance level upon which we decide to reject the null hypothesis :math:`H_0`. Suppose we choose the common significance level of 5%. Our alternative hypothesis is that college professors are truly more satisfied with their work than scientists. Therefore, we expect :math:`p_1` the proportion of very satisfied college professors to be greater than :math:`p_2`, the proportion of very satisfied scientists. We thus call `boschloo_exact` with the ``alternative="greater"`` option: >>> import scipy.stats as stats >>> res = stats.boschloo_exact([[74, 31], [43, 32]], alternative="greater") >>> res.statistic 0.0483... >>> res.pvalue 0.0355... Under the null hypothesis that scientists are happier in their work than college professors, the probability of obtaining test results at least as extreme as the observed data is approximately 3.55%. Since this p-value is less than our chosen significance level, we have evidence to reject :math:`H_0` in favor of the alternative hypothesis. r r r� r r r r( r r& rc r� r\ )r^ rL r% r$ z `alternative` should be one of )r\ rc r� z, found r r g� �?r r r! )r � hypergeomr/ r, r- r' r� r5 rm r; r� ri r8 r� r< r r r) r r r r( r� r* )r r^ rL r= r, r- �totalr{ r| r5 �pvalues� boschloo_less�boschloo_greaterr� r r4 � fisher_statr3 r6 r7 r8 r9 r: s rZ r r � s � �F �'�I��A�v�v�� �� � � � � � �J�u�B�H� -� -� -�E� �;�&� � ��E�F�F�F� �v�e�a�i��� G��E�F�F�F��E�I�I�1�I� � � � � #�2�6�2�6�2�2�2� $�y�y�a�y� 0� 0� �K�� �+� %�E� ��;��?�"�(� 3� 3� 3� ;� ;�A�r� B� B�B� ��;��?�"�(� 3� 3� 3� ;� ;�B�� B� B�B��R��I��f� � ��-�-��E�9�k�B�B�D��� � � !� !��-�-��E�9�k�B�B�D��� � � #� #�&�u�&�A�F�F�F� �)�%�Y�!�L�L�L�� +� 1�4D�4K� K� K�M�M�!� � ���S�Z� �q��:�:�:��"�3�=�&�9�9�9� &�-M� &� &�!� &� &� � ��o�o� ��%��+�u�T�{�2�3�K� �;�'�2�2�I���b�d�I�K�I��B�+�K�8�8�K�+�K�8�8�K�$�R� �;�r�?�:�� �5��+�Y� 7�� � � � � �F� �g�b�f�f�j�[�)�)��!�<�<�<�G� �{�G� 4� 4�4r[ c �$ � |j \ }}||z dz }t j dd�� � 5 t j | t j | � � | dk �� � }t j d| z t j | � � d| z dk �� � }||z } d| |dk dd�dd�f < |||z z } d| ||k dd�dd�f < || z | z } ddd� � n # 1 swxY w Y | | } | � � � } t j dd�� � 5 t j | | z � � � � � }| t j |t j |t j � � |dk �� � z }ddd� � n # 1 swxY w Y | S )a* Compute the log pvalue in respect of a nuisance parameter considering a 2x2 sample space. Parameters ---------- nuisance_param : float nuisance parameter used in the computation of the maximisation of the p-value. Must be between 0 and 1 x1_sum_x2 : ndarray Sum of x1 and x2 inside barnard_exact x1_sum_x2_log_comb : ndarray sum of the log combination of x1 and x2 index_arr : ndarray of boolean Returns ------- p_value : float Return the maximum p-value considering every nuisance paramater between 0 and 1 Notes ----- Both Barnard's test and Boschloo's test iterate over a nuisance parameter :math:`\pi \in [0, 1]` to find the maximum p-value. To search this maxima, this function return the negative log pvalue with respect to the nuisance parameter passed in params. This negative log p-value is then used in `shgo` to find the minimum negative pvalue which is our maximum pvalue. Also, to compute the different combination used in the p-values' computation formula, this function uses `gammaln` which is more tolerant for large value than `scipy.special.comb`. `gammaln` gives a log combination. For the little precision loss, performances are improved a lot. r% r` r r )�out�wherer N) r� r, rk �logr� rE r� rm � full_like�inf)�nuisance_paramr5 r8 r3 �t1�t2rL � log_nuisance�log_1_minus_nuisance�nuisance_power_x1_x2� nuisance_power_n_minus_x1_x2�tmp_log_values_arr�tmp_values_from_index� max_value� log_probs� log_pvalues rZ r( r( m sW � �V �_�F�B�� �R��!� �A� ��H�h� 7� 7� 7� � ��v� �� �n�-�-� �A�%� � � � � "�v� �� �� �n�-�-��n�$��)� � � � � ,�i� 7� �78� �i�1�n�a�a�a����d� 3�4�';�q�9�}�'M�$�?@�$�i�1�n�a�a�a����d�%;�<� �"� #�*� +� �% � � � � � � � � � � ���� � � � �0 /�y� 9�� &�)�)�+�+�I� ��H�h� 7� 7� 7� � ��F�0�9�<�=�=�A�A�C�C� � ��� �� �Y����0�0��a�-�" � " � " � � � � � � � � � � � � � ���� � � � � �;�s% �B C�C� C�A&F�F� Fc � �� t j ||� � }||z }||z }||z }|dz ||z z d| z |d|z dz z z z d|dz z z }|dz ||z z }t ||z |� � } t || � � } t j | � � �t j dgdgg��� � g�fd�t |� � D � � z } t |dz � � D ]�} g } t j d��� � }t | � � D ]�\ }}t j |d |d d � � � \ }}}t j t j ||d|f |d|f z g� � t j ||d� � t j ||d� � gd� � }|dxx ||z || z z dz z cc< | � |� � ��| } ��| | \ }}t j t j |||k � � | z � � S ) u Compute the exact p-value of the Cramer-von Mises two-sample test for a given value s of the test statistic. m and n are the sizes of the samples. [1] Y. Xiao, A. Gordon, and A. Yakovlev, "A C++ Program for the Cramér-Von Mises Two-Sample Test", J. Stat. Soft., vol. 17, no. 8, pp. 1-15, Dec. 2006. [2] T. W. Anderson "On the Distribution of the Two-Sample Cramer-von Mises Criterion," The Annals of Mathematical Statistics, Ann. Math. Statist. 33(3), 1148-1159, (September, 1962) r% � r� r r r� c �< �� g | ]}t j d ��� � ��S )�r% r r� )r, �empty)�.0r r� s �rZ � <listcomp>z)_pval_cvm_2samp_exact.<locals>.<listcomp>� s( �� � =� =� =�q�R�X�f�E� *� *� *� =� =� =r[ rX T)�return_indices)r, �lcmr rE �min_scalar_typer r� rY � enumerate� intersect1d� concatenate�stack�delete�append�float64rm )�sr� rL r] �ar� �mn�zeta� zeta_boundr �max_gs�gsr� �next_gs�tmp�v�g�vi�i0�i1�value�freqr� s @rZ �_pval_cvm_2samp_exactrv � sA �� � �&��A�,�,�C� �q��A� �q��A� �Q��B� �!�8�q�1�u� ��Q���q�2�v��z�):�!:� ;��B�!�G� � L�D� �a��1�q�5�!�J���A��q�>�>�L� ��\� *� *�F� � �v� &� &�E� �8�a�S�1�#�J�e� ,� ,� ,� -� =� =� =� =�E�!�H�H� =� =� =� >�B� �1�q�5�\�\� � �����h�v�U�+�+�+�� �b�M�M� � �D�A�q� � ��A���!��T�J�J�J�J�B��B��.���"�c�!�R�%�j�1�Q��U�8� 3�4�5�5�� �#�r�1�%�%�� �!�R��#�#�"� � � �C� ��F�F�F�q�1�u�q�1�u�}��*� *�F�F�F� �N�N�3� � � � � ����Q�%�K�E�4� �:�b�f�T�%�4�-� 0�1�1�L�@� A� A�Ar[ �autoc �� � t j t j | � � � � }t j t j |� � � � }|j dk s |j dk rt d� � �|j dk s |j dk rt d� � �|dvrt d� � �t |� � }t |� � }|dk rt ||� � dk rd}nd }t j ||g� � }t j � |d � � � }|d |� } ||d � } |t j | t j d|dz � � z d z � � z } | |t j | t j d|dz � � z d z � � z z } ||z ||z } } | | | z z d| z dz d| z z z }|d k rt | ||� � }n�dd| z z dz }| dz d| z | z d|d z |d z z z z d | z z z }|d| d z z dz | z z }d||z t j d|z � � z z }|dk rd}n t ddt! |� � z � � }t# ||�� � S )u� Perform the two-sample Cramér-von Mises test for goodness of fit. This is the two-sample version of the Cramér-von Mises test ([1]_): for two independent samples :math:`X_1, ..., X_n` and :math:`Y_1, ..., Y_m`, the null hypothesis is that the samples come from the same (unspecified) continuous distribution. Parameters ---------- x : array_like A 1-D array of observed values of the random variables :math:`X_i`. y : array_like A 1-D array of observed values of the random variables :math:`Y_i`. method : {'auto', 'asymptotic', 'exact'}, optional The method used to compute the p-value, see Notes for details. The default is 'auto'. Returns ------- res : object with attributes statistic : float Cramér-von Mises statistic. pvalue : float The p-value. See Also -------- cramervonmises, anderson_ksamp, epps_singleton_2samp, ks_2samp Notes ----- .. versionadded:: 1.7.0 The statistic is computed according to equation 9 in [2]_. The calculation of the p-value depends on the keyword `method`: - ``asymptotic``: The p-value is approximated by using the limiting distribution of the test statistic. - ``exact``: The exact p-value is computed by enumerating all possible combinations of the test statistic, see [2]_. If ``method='auto'``, the exact approach is used if both samples contain equal to or less than 20 observations, otherwise the asymptotic distribution is used. If the underlying distribution is not continuous, the p-value is likely to be conservative (Section 6.2 in [3]_). When ranking the data to compute the test statistic, midranks are used if there are ties. References ---------- .. [1] https://en.wikipedia.org/wiki/Cramer-von_Mises_criterion .. [2] Anderson, T.W. (1962). On the distribution of the two-sample Cramer-von-Mises criterion. The Annals of Mathematical Statistics, pp. 1148-1159. .. [3] Conover, W.J., Practical Nonparametric Statistics, 1971. Examples -------- Suppose we wish to test whether two samples generated by ``scipy.stats.norm.rvs`` have the same distribution. We choose a significance level of alpha=0.05. >>> import numpy as np >>> from scipy import stats >>> rng = np.random.default_rng() >>> x = stats.norm.rvs(size=100, random_state=rng) >>> y = stats.norm.rvs(size=70, random_state=rng) >>> res = stats.cramervonmises_2samp(x, y) >>> res.statistic, res.pvalue (0.29376470588235293, 0.1412873014573014) The p-value exceeds our chosen significance level, so we do not reject the null hypothesis that the observed samples are drawn from the same distribution. For small sample sizes, one can compute the exact p-values: >>> x = stats.norm.rvs(size=7, random_state=rng) >>> y = stats.t.rvs(df=2, size=6, random_state=rng) >>> res = stats.cramervonmises_2samp(x, y, method='exact') >>> res.statistic, res.pvalue (0.197802197802198, 0.31643356643356646) The p-value based on the asymptotic distribution is a good approximation even though the sample size is small. >>> res = stats.cramervonmises_2samp(x, y, method='asymptotic') >>> res.statistic, res.pvalue (0.197802197802198, 0.2966041181527128) Independent of the method, one would not reject the null hypothesis at the chosen significance level in this example. r z/x and y must contain at least two observations.z$The samples must be one-dimensional.)rw �exact� asymptoticz0method must be either auto, exact or asymptotic.rw � rz ry �average)�methodNr% r� rV r� �- gUUUUUU�?g�~j�t�h?r+ r r )r, r� r- r� r/ r. r1 rE ra r� r� �rankdatarm ri rv rf r� r� )rG rH r} �xa�yarJ rK r� rU �rx�ryr� r� �NrI rY �et�vt�tns rZ r r � s� � �B ����A� � � � �B� ����A� � � � �B� �w�!�|�|�r�w�!�|�|��J�K�K�K� �w��{�{�b�g��k�k��?�@�@�@� �4�4�4��K�L�L�L� �R���B� �R���B� ���� �r�2�;�;�� � �!�F�F� �F� ���B�x� � �A� � � � �Q�y� �1�1�A� �3�B�3��B� �2�3�3��B� �R�V�R�"�)�A�r�!�t�,�,�,�q�0� 1� 1�1�A���b�f�b�2�9�Q��1��-�-�-��1�2�2� 2�2�A� �b�5�"�r�'�q�A� �Q�q�S� �Q�q�S�1�W�q��s�O�#�A� ���� !�!�R�� ,� ,��� �!�A�#�g�q�[����c�a��c�!�e�a��Q���Q�� � /�/�!�A�#�5� 6�� �2��1��9�q�=�1�$� %�� �A��F�b�g�b�2�g� .� .�.� .�� ��:�:��A�A��A�r�L�� ,� ,�,�-�-�A� �!�A� 6� 6� 6�6r[ c �& � e Zd ZdZd� Zd� Zdd�ZdS )�TukeyHSDResulta� Result of `scipy.stats.tukey_hsd`. Attributes ---------- statistic : float ndarray The computed statistic of the test for each comparison. The element at index ``(i, j)`` is the statistic for the comparison between groups ``i`` and ``j``. pvalue : float ndarray The associated p-value from the studentized range distribution. The element at index ``(i, j)`` is the p-value for the comparison between groups ``i`` and ``j``. Notes ----- The string representation of this object displays the most recently calculated confidence interval, and if none have been previously calculated, it will evaluate ``confidence_interval()``. References ---------- .. [1] NIST/SEMATECH e-Handbook of Statistical Methods, "7.4.7.1. Tukey's Method." https://www.itl.nist.gov/div898/handbook/prc/section4/prc471.htm, 28 November 2020. c �h � || _ || _ || _ || _ || _ d | _ d | _ d S r� )r r � _ntreatments�_nobs� _stand_err�_ci�_ci_cl)r� r r r� r� r� s rZ r� zTukeyHSDResult.__init__� s8 � �"��� �� � (����� �$�������� � � r[ c � � | j �| � d�� � d| j dz d�d�}|dz }t | j j d � � D ]�}t | j j d � � D ]_}||k rW|d |� d |� d | j ||f d �| j ||f d �| j j ||f d �| j j ||f d �d � z }�`��|S )N�ffffff�?)�confidence_levelz(Tukey's HSD Pairwise Group Comparisons (�d z.1fz% Confidence Interval) z3Comparison Statistic p-value Lower CI Upper CI r z (z - z) z>10.3f� ) r� �confidence_intervalr� r� r r� r �low�high)r� rf �ir� s rZ �__str__zTukeyHSDResult.__str__� s: � � �8� � � $� $�c� $� :� :� :�@��+�c�/� ?�@� @� @�� � C�C���t�{�(��+�,�,� =� =�A��4�;�,�Q�/�0�0� =� =����6�6�� <�q� <� <�Q� <� <�$�.��A��*>�G� <� !�[��A�� .�7�<� !�X�\�!�Q�$� /�8�<� "�X�]�1�a�4� 0�;�<� <� <� =�A�� =� �r[ r� c �V � | j �| j �|| j k r| j S d|cxk rdk sn t d� � �|| j | j | j z f}t j j |� }|| j z }| j |z }| j |z }t ||�� � | _ || _ | j S )a3 Compute the confidence interval for the specified confidence level. Parameters ---------- confidence_level : float, optional Confidence level for the computed confidence interval of the estimated proportion. Default is .95. Returns ------- ci : ``ConfidenceInterval`` object The object has attributes ``low`` and ``high`` that hold the lower and upper bounds of the confidence intervals for each comparison. The high and low values are accessible for each comparison at index ``(i, j)`` between groups ``i`` and ``j``. References ---------- .. [1] NIST/SEMATECH e-Handbook of Statistical Methods, "7.4.7.1. Tukey's Method." https://www.itl.nist.gov/div898/handbook/prc/section4/prc471.htm, 28 November 2020. Examples -------- >>> from scipy.stats import tukey_hsd >>> group0 = [24.5, 23.5, 26.4, 27.1, 29.9] >>> group1 = [28.4, 34.2, 29.5, 32.2, 30.1] >>> group2 = [26.1, 28.3, 24.3, 26.2, 27.8] >>> result = tukey_hsd(group0, group1, group2) >>> ci = result.confidence_interval() >>> ci.low array([[-3.649159, -8.249159, -3.909159], [ 0.950841, -3.649159, 0.690841], [-3.389159, -7.989159, -3.649159]]) >>> ci.high array([[ 3.649159, -0.950841, 3.389159], [ 8.249159, 3.649159, 7.989159], [ 3.909159, -0.690841, 3.649159]]) Nr r z)Confidence level must be between 0 and 1.)r� r� ) r� r� r/ r� r� r �studentized_rangerh r� r r )r� r� �params�srd�tukey_criterion� upper_conf� lower_confs rZ r� z"TukeyHSDResult.confidence_interval� s� � �V �H� �T�[�%<� �D�K�/�/��8�O��#�'�'�'�'�a�'�'�'�'� � H�I�I� I� #�D�J��0A�D�J�0N�O���-�1�6�:�� ���/�� �^�o�5� ��^�o�5� �%�*�:�F�F�F���&�� ��x�r[ N)r� )r� r� r� �__doc__r� r� r� r� r[ rZ r� r� � sV � � � � � �� �6� � �� � �$C� C� C� C� C� Cr[ r� c �> � t | � � dk rt d� � �d� | D � � } | D ]k}|j dk rt d� � �|j dk rt d� � �t j |� � � � � rt d� � ��l| S )Nr% z$There must be more than 1 treatment.c �6 � g | ]}t j |� � ��S r� )r, r- �rZ �args rZ r[ z!_tukey_hsd_iv.<locals>.<listcomp> s � � ,� ,� ,��B�J�s�O�O� ,� ,� ,r[ r z&Input samples must be one-dimensional.z+Input sample size must be greater than one.z Input samples must be finite.)r1 r/ r. r� r, �isinfr5 )r� r� s rZ � _tukey_hsd_ivr� s� � � �D� � �Q����?�@�@�@� ,� ,�t� ,� ,� ,�D�� >� >�� �8�q�=�=� � E�F�F� F� �8�q�=�=� � J�K�K� K� �8�C�=�=� � � � � >� � <�=�=� =� >� �Kr[ c � � t | � � } t | � � }t j d� | D � � � � }t j d� | D � � � � }t j |� � }t j d� | D � � |dz z � � ||z z }t j |� � j dk r d|d z }nd|z d|d j z z }t j ||z dz � � }|d j |z }t j |� � |z } | |||z f} t j j | � } t || |||� � S )a3 Perform Tukey's HSD test for equality of means over multiple treatments. Tukey's honestly significant difference (HSD) test performs pairwise comparison of means for a set of samples. Whereas ANOVA (e.g. `f_oneway`) assesses whether the true means underlying each sample are identical, Tukey's HSD is a post hoc test used to compare the mean of each sample to the mean of each other sample. The null hypothesis is that the distributions underlying the samples all have the same mean. The test statistic, which is computed for every possible pairing of samples, is simply the difference between the sample means. For each pair, the p-value is the probability under the null hypothesis (and other assumptions; see notes) of observing such an extreme value of the statistic, considering that many pairwise comparisons are being performed. Confidence intervals for the difference between each pair of means are also available. Parameters ---------- sample1, sample2, ... : array_like The sample measurements for each group. There must be at least two arguments. Returns ------- result : `~scipy.stats._result_classes.TukeyHSDResult` instance The return value is an object with the following attributes: statistic : float ndarray The computed statistic of the test for each comparison. The element at index ``(i, j)`` is the statistic for the comparison between groups ``i`` and ``j``. pvalue : float ndarray The computed p-value of the test for each comparison. The element at index ``(i, j)`` is the p-value for the comparison between groups ``i`` and ``j``. The object has the following methods: confidence_interval(confidence_level=0.95): Compute the confidence interval for the specified confidence level. Notes ----- The use of this test relies on several assumptions. 1. The observations are independent within and among groups. 2. The observations within each group are normally distributed. 3. The distributions from which the samples are drawn have the same finite variance. The original formulation of the test was for samples of equal size [6]_. In case of unequal sample sizes, the test uses the Tukey-Kramer method [4]_. References ---------- .. [1] NIST/SEMATECH e-Handbook of Statistical Methods, "7.4.7.1. Tukey's Method." https://www.itl.nist.gov/div898/handbook/prc/section4/prc471.htm, 28 November 2020. .. [2] Abdi, Herve & Williams, Lynne. (2021). "Tukey's Honestly Significant Difference (HSD) Test." https://personal.utdallas.edu/~herve/abdi-HSD2010-pretty.pdf .. [3] "One-Way ANOVA Using SAS PROC ANOVA & PROC GLM." SAS Tutorials, 2007, www.stattutorials.com/SAS/TUTORIAL-PROC-GLM.htm. .. [4] Kramer, Clyde Young. "Extension of Multiple Range Tests to Group Means with Unequal Numbers of Replications." Biometrics, vol. 12, no. 3, 1956, pp. 307-310. JSTOR, www.jstor.org/stable/3001469. Accessed 25 May 2021. .. [5] NIST/SEMATECH e-Handbook of Statistical Methods, "7.4.3.3. The ANOVA table and tests of hypotheses about means" https://www.itl.nist.gov/div898/handbook/prc/section4/prc433.htm, 2 June 2021. .. [6] Tukey, John W. "Comparing Individual Means in the Analysis of Variance." Biometrics, vol. 5, no. 2, 1949, pp. 99-114. JSTOR, www.jstor.org/stable/3001913. Accessed 14 June 2021. Examples -------- Here are some data comparing the time to relief of three brands of headache medicine, reported in minutes. Data adapted from [3]_. >>> import numpy as np >>> from scipy.stats import tukey_hsd >>> group0 = [24.5, 23.5, 26.4, 27.1, 29.9] >>> group1 = [28.4, 34.2, 29.5, 32.2, 30.1] >>> group2 = [26.1, 28.3, 24.3, 26.2, 27.8] We would like to see if the means between any of the groups are significantly different. First, visually examine a box and whisker plot. >>> import matplotlib.pyplot as plt >>> fig, ax = plt.subplots(1, 1) >>> ax.boxplot([group0, group1, group2]) >>> ax.set_xticklabels(["group0", "group1", "group2"]) # doctest: +SKIP >>> ax.set_ylabel("mean") # doctest: +SKIP >>> plt.show() From the box and whisker plot, we can see overlap in the interquartile ranges group 1 to group 2 and group 3, but we can apply the ``tukey_hsd`` test to determine if the difference between means is significant. We set a significance level of .05 to reject the null hypothesis. >>> res = tukey_hsd(group0, group1, group2) >>> print(res) Tukey's HSD Pairwise Group Comparisons (95.0% Confidence Interval) Comparison Statistic p-value Lower CI Upper CI (0 - 1) -4.600 0.014 -8.249 -0.951 (0 - 2) -0.260 0.980 -3.909 3.389 (1 - 0) 4.600 0.014 0.951 8.249 (1 - 2) 4.340 0.020 0.691 7.989 (2 - 0) 0.260 0.980 -3.389 3.909 (2 - 1) -4.340 0.020 -7.989 -0.691 The null hypothesis is that each group has the same mean. The p-value for comparisons between ``group0`` and ``group1`` as well as ``group1`` and ``group2`` do not exceed .05, so we reject the null hypothesis that they have the same means. The p-value of the comparison between ``group0`` and ``group2`` exceeds .05, so we accept the null hypothesis that there is not a significant difference between their means. We can also compute the confidence interval associated with our chosen confidence level. >>> group0 = [24.5, 23.5, 26.4, 27.1, 29.9] >>> group1 = [28.4, 34.2, 29.5, 32.2, 30.1] >>> group2 = [26.1, 28.3, 24.3, 26.2, 27.8] >>> result = tukey_hsd(group0, group1, group2) >>> conf = res.confidence_interval(confidence_level=.99) >>> for ((i, j), l) in np.ndenumerate(conf.low): ... # filter out self comparisons ... if i != j: ... h = conf.high[i,j] ... print(f"({i} - {j}) {l:>6.3f} {h:>6.3f}") (0 - 1) -9.480 0.280 (0 - 2) -5.140 4.620 (1 - 0) -0.280 9.480 (1 - 2) -0.540 9.220 (2 - 0) -4.620 5.140 (2 - 1) -9.220 0.540 c �6 � g | ]}t j |� � ��S r� )r, rC r� s rZ r[ z tukey_hsd.<locals>.<listcomp>� s � �5�5�5����� � �5�5�5r[ c � � g | ] }|j �� S r� )r� )rZ rg s rZ r[ z tukey_hsd.<locals>.<listcomp>� s � �%;�%;�%;��a�f�%;�%;�%;r[ c �: � g | ]}t j |d �� � ��S )r )�ddof)r, rs r� s rZ r[ z tukey_hsd.<locals>.<listcomp>� s' � �7�7�7�3�2�6�#�A�&�&�&�7�7�7r[ r r% r N)r� r1 r, r- rm �uniquer� r< rf rl r r� rF r� ) r� � ntreatments�means�nsamples_treatments�nobs�mse� normalize� stand_err�mean_differences�t_statr� r? s rZ r r s{ � �` �� � �D��d�)�)�K� �J�5�5��5�5�5� 6� 6�E� �*�%;�%;�d�%;�%;�%;�<�<�� �6�%� &� &�D� �6�7�7�$�7�7�7�&��*�,� -� -�04�{�0B� D�C� �y�$�%�%�*�a�/�/� �+�A�.�.� � � �+�+�a�2E�d�2K�2M�.M�M� � �� �C��!�+�,�,�I� �T�{�}�u�,�� �V�$� %� %� � 1�F� �[�$��"4� 4�F��-�0�&�9�G� �*�G�[� � � +� +� +r[ )r! r� )r� )r\ )Nr\ )r\ Tr )r\ r )rw )@� collectionsr � dataclassesr �mathr �numpyr, rA � itertoolsr r6 r� �scipy.optimizer � r �_commonr �_continuous_distnsr r � scipy.specialr r r � scipy.fftr �_stats_pythranr r r� r r� r �__all__r r r rd r� r� r� r� r r� r� r� r r r r r r� r r r; r r( rv r r� r� r r� r[ rZ �<module>r� s� �� "� "� "� "� "� "� &� &� &� &� &� &� � � � � � � � � � � ���� "� "� "� "� "� "� � � � � � � � � � � � � � � � � '� '� '� '� '� '� *� *� *� *� *� *� *� *� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� � � � � � � *� *� *� *� *� *�� � � � � � � � "� !� !� !� !� !� .� .� .�� (�Z�(D�(?� A� A� �v,� v,� v,� v,�r 01�k� v8� v8� v8� v8� v8�rJ� J� J�(*� *� *� *� *� *� *� *�/� /� /�d!� !� !�H � � � �Br7� r7� r7� r7�j � � �"� � �6� � �6� � � �4 ��� ?�A� A� �r� r� r� r�l � � � <� <� <� $�^��K��/�(�E�1B�C�� � � F6� F6� F6� F6�R %�n��[�%� 0�8�U�2C�D�� � � N5� N5� N5� N5�bX� X� X�v1B� 1B� 1B�hV7� V7� V7� V7�rz� z� z� z� z� z� z� z�z � � � v+� v+� v+� v+� v+r[
Save File
Cancel