+ Zjw ^RIt^RIt^RIHtHt^RIHt^RI H t H t H t H t HtHtHt^RIHtHtHtHt^RIHt^RIHtHtHt.R#OtRtR tR tR t ] !R$.R 7]!]R ^] ^^R7R^RRRR/Rl44t!] !R R.R R%.RR7]!]R ^] ^^R7RRRRR^/Rl44t"] !R$R&.R R7]!]R^]RR ^R7R'RR/Rll44t#] !R$R(.R R7]!]R^]RR ^R7R)RR/R ll44t$R!R/R"lt%R#)*N)statsspecial)array_api_extra)xp_capabilitiesarray_namespace xp_promoteis_numpy _share_masks_count_nonmasked is_marray) _SimpleNormalSignificanceResult _get_pvalue _rankdata)_axis_nan_policy_factory)TheilslopesResult_n_samples_optional_xSiegelslopesResultc DVPR,pVPVRR7pVPWQ4wrQVPWRR7p\P !VRRR7p\P !V)RRR7pVP VPVPVRR74RR7pV'd'^^V,V^,^, , , p M=^VP WG, V,RR7,p ^WH,V , , p WV3#)axismax)methodr) shapeargsortbroadcast_arraystake_along_axisrrankdatasumabsdiff) xy y_continuousxpnjrlnum statisticdens &&&& Y/home/luca/screen-checking/.venv/lib/python3.14/site-packages/scipy/stats/_correlation.py _xi_statisticr0s  A 12 A  q $DA 1b)A qR0A r%b1A &&+,2& 6CC16A:.. "&&!%12&..# % ?c@VPR,pV'dMVP\P!R4\P!V4, VPR7#VP ^V^,VPR7pVP VRR7pVPVRR7p^V^,, VP^V,^V,, ^,V^,,RR7,p^V^,, VPWtV, V,,^,RR7,p ^V^,, VP^V,^V,, ^,V,RR7,p ^V^,, VPWV, ,RR7,p V^V ,, V ^,,V ^,, p VPV 4\P!V4, #)rdtyperrg?) rasarraymathsqrtr4arangesortcumulative_sumr!) r*r+r&r'r(iuvanbncndntau2s &&&& r/_xi_stdrC-s  Azz$))E*TYYq\9zII !QU!'' *A A !"%A QTBFFAaC!A#IMQT1F; ;B QTBFFAQ MA-BF7 7B QTBFFAaC!A#IMQ.RF8 8B QTBFFAQKrF2 2B 2IA Q &D 774=499Q< ''r1cVR9d \R4h\V\P4'g%VP 4pRpVR8wd \V4hW3#)Tz`y_continuous` must be boolean.z@`method` must be 'asymptotic' or a `PermutationMethod` instance. asymptotic>FT) ValueError isinstancerPermutationMethodlower)r&rmessages&& r/_chatterjeexi_ivrKGsW=(:;; fe55 6 6T \ !W% %  r1c2VPVP3#N)r-pvalue)res_s&&r/_unpackrQWs ==#** $$r1) skip_backendsT)paired n_samplesresult_to_tuple n_outputs too_smallrr&FrrEc Taaa \SV4o \SV4wop\SVRS R7wopRpVR8Xd@\SVSS R7wrgp\ WxSS R7p \ 4p \ Wi, WS R7p Mn\V\P4'dO\P!RRV3RVV V3R lR VR R /VP4BR R/Bp V PV PrXP^8Xd VR,MTpX P^8Xd V R,MT p \Wk4#)aJCompute the xi correlation and perform a test of independence. The xi correlation coefficient is a measure of association between two variables; the value tends to be close to zero when the variables are independent and close to 1 when there is a strong association. Unlike other correlation coefficients, the xi correlation is effective even when the association is not monotonic. Parameters ---------- x, y : array-like The samples: corresponding observations of the independent and dependent variable. The (N-d) arrays must be broadcastable. axis : int, default: 0 Axis along which to perform the test. y_continuous : bool, default: False Whether `y` is assumed to be drawn from a continuous distribution. If `y` is drawn from a continuous distribution, results are valid whether this is assumed or not, but enabling this assumption will result in faster computation and typically produce similar results. method : 'asymptotic' or `PermutationMethod` instance, optional Selects the method used to calculate the *p*-value. Default is 'asymptotic'. The following options are available. * ``'asymptotic'``: compares the standardized test statistic against the normal distribution. * `PermutationMethod` instance. In this case, the p-value is computed using `permutation_test` with the provided configuration options and other appropriate settings. Returns ------- res : SignificanceResult An object containing attributes: statistic : float The xi correlation statistic. pvalue : float The associated *p*-value: the probability of a statistic at least as high as the observed value under the null hypothesis of independence. See Also -------- scipy.stats.pearsonr, scipy.stats.spearmanr, scipy.stats.kendalltau Notes ----- There is currently no special handling of ties in `x`; they are broken arbitrarily by the implementation. [1]_ recommends: "if there are ties among the Xi's, then choose an increasing rearrangement as above by breaking ties uniformly at random." This is easily accomplished by adding a small amount of random noise to `x`; see examples. [1]_ notes that the statistic is not symmetric in `x` and `y` *by design*: "...we may want to understand if :math:`Y` is a function :math:`X`, and not just if one of the variables is a function of the other." See [1]_ Remark 1. References ---------- .. [1] Chatterjee, Sourav. "A new coefficient of correlation." Journal of the American Statistical Association 116.536 (2021): 2009-2022. :doi:`10.1080/01621459.2020.1758115`. Examples -------- Generate perfectly correlated data, and observe that the xi correlation is nearly 1.0. >>> import numpy as np >>> from scipy import stats >>> rng = np.random.default_rng(348932549825235) >>> x = rng.uniform(0, 10, size=100) >>> y = np.sin(x) >>> res = stats.chatterjeexi(x, y) >>> res.statistic np.float64(0.9012901290129013) The probability of observing such a high value of the statistic under the null hypothesis of independence is very low. >>> res.pvalue np.float64(2.2206974648177804e-46) As noise is introduced, the correlation coefficient decreases. >>> noise = rng.normal(scale=[[0.1], [0.5], [1]], size=(3, 100)) >>> res = stats.chatterjeexi(x, y + noise, axis=-1) >>> res.statistic array([0.79507951, 0.41824182, 0.16651665]) Because the distribution of `y` is continuous, it is valid to pass ``y_continuous=True``. The statistic is identical, and the p-value (not shown) is only slightly different. >>> stats.chatterjeexi(x, y + noise, y_continuous=True, axis=-1).statistic array([0.79507951, 0.41824182, 0.16651665]) Consider a case in which there are ties in `x`. >>> x = rng.integers(10, size=1000) >>> y = rng.integers(10, size=1000) [1]_ recommends breaking the ties uniformly at random. >>> d = rng.uniform(1e-5, size=x.size) >>> res = stats.chatterjeexi(x + d, y) >>> res.statistic -0.029919991638798438 Since this gives a randomized estimate of the statistic, [1]_ also suggests considering the average over all possibilities of breaking ties. This is computationally infeasible when there are many ties, but a randomized estimate of *this* quantity can be obtained by considering many random possibilities of breaking ties. >>> d = rng.uniform(1e-5, size=(9999, x.size)) >>> res = stats.chatterjeexi(x + d, y, axis=1) >>> np.mean(res.statistic) 0.001186895213756626 Tforce_floatingr'greaterrEr') alternativer'datar-c0<\SVSSR7^,#)r\)r0)r%rr$r'r&s&&r/chatterjeexi..smAq,2&Nq&Qr1r]permutation_typepairingsrr)rrKrr0rCr rrGrrHpermutation_test_asdictr-rNndimr)r$r%rr&rr]xir*r+stdnormrNrOr'sf&$d$ @r/ chatterjeexirk[s2z A B,L&AL& a4B 7DAq K  A|;qaLR0RXtL FE33 4 4$$R$ 7A EKNNDT   ]]CJJF77a<BRB!;;!+VBZF b ))r1 jax.numpyz:Only the default `method` is compatible with MArray input.)cpu_only exceptionsmarrayrR extra_noter]z two-sidedc\W4p\WVR7wr\P!WR7p\P!WR7p\P!WgW2VR7p\ VP VP4#)a'Calculate a Spearman rho correlation coefficient with associated p-value. The Spearman rank-order correlation coefficient is a nonparametric measure of the monotonicity of the relationship between two datasets. Like other correlation coefficients, it varies between -1 and +1 with 0 implying no correlation. Coefficients of -1 or +1 are associated with an exact monotonic relationship. Positive correlations indicate that as `x` increases, so does `y`; negative correlations indicate that as `x` increases, `y` decreases. The p-value is the probability of an uncorrelated system producing datasets with a Spearman correlation at least as extreme as the one computed from the observed dataset. Parameters ---------- x, y : array-like The samples: corresponding observations of the independent and dependent variable. The (N-d) arrays must be broadcastable. alternative : {'two-sided', 'less', 'greater'}, optional Defines the alternative hypothesis. Default is 'two-sided'. The following options are available: * 'two-sided': the correlation is nonzero * 'less': the correlation is negative (less than zero) * 'greater': the correlation is positive (greater than zero) method : ResamplingMethod, optional Defines the method used to compute the p-value. If `method` is an instance of `PermutationMethod`/`MonteCarloMethod`, the p-value is computed using `scipy.stats.permutation_test`/`scipy.stats.monte_carlo_test` with the provided configuration options and other appropriate settings. Otherwise, the p-value is computed using an asymptotic approximation of the null distribution. axis : int or None, optional If axis=0 (default), then each column represents a variable, with observations in the rows. If axis=1, the relationship is transposed: each row represents a variable, while the columns contain observations. If axis=None, then both arrays will be raveled. Like other `scipy.stats` functions, `axis` is interpreted after the arrays are broadcasted. Returns ------- res : SignificanceResult An object containing attributes: statistic : floating point array or NumPy scalar Spearman correlation coefficient pvalue : floating point array NumPy scalar The p-value - the probabilitiy of realizing such an extreme statistic value under the null hypothesis that two samples have no ordinal correlation. See `alternative` above for alternative hypotheses. Warns ----- `~scipy.stats.ConstantInputWarning` Raised if an input is a constant array. The correlation coefficient is not defined in this case, so ``np.nan`` is returned. Notes ----- `spearmanrho` was created to make improvements to SciPy's implementation of the Spearman correlation test without making backward-incompatible changes to `spearmanr`. Advantages of `spearmanrho` over `spearmanr` include: - `spearmanrho` follows standard array broadcasting rules. - `spearmanrho` is compatible with some non-NumPy arrays. - `spearmanrho` can compute exact p-values, even in the presence of ties, when an appropriate instance of `PermutationMethod` is provided via the `method` argument. References ---------- .. [1] Zwillinger, D. and Kokoska, S. (2000). CRC Standard Probability and Statistics Tables and Formulae. Chapman & Hall: New York. 2000. Section 14.7 .. [2] Kendall, M. G. and Stuart, A. (1973). The Advanced Theory of Statistics, Volume 2: Inference and Relationship. Griffin. 1973. Section 31.18 Examples -------- Univariate samples, approximate p-value. >>> import numpy as np >>> from scipy import stats >>> x = [1, 2, 3, 4, 5] >>> y = [5, 6, 7, 8, 7] >>> res = stats.spearmanrho(x, y) >>> res.statistic np.float64(0.8207826816681233) >>> res.pvalue np.float64(0.08858700531354405) Univariate samples, exact p-value. >>> res = stats.spearmanrho(x, y, method=stats.PermutationMethod()) >>> res.statistic np.float64(0.8207826816681233) >>> res.pvalue np.float64(0.13333333333333333) Batch of univariate samples, one vectorized call. >>> rng = np.random.default_rng(98145152315484) >>> x2 = rng.standard_normal((2, 100)) >>> y2 = rng.standard_normal((2, 100)) >>> res = stats.spearmanrho(x2, y2, axis=-1) >>> res.statistic array([ 0.16585659, -0.12151215]) >>> res.pvalue array([0.0991155 , 0.22846869]) Bivariate samples using standard broadcasting rules. >>> res = stats.spearmanrho(x2[np.newaxis, :], x2[:, np.newaxis], axis=-1) >>> res.statistic array([[ 1. , -0.14670267], [-0.14670267, 1. ]]) >>> res.pvalue array([[0. , 0.14526128], [0.14526128, 0. ]]) r\r)rr]r)rr rr pearsonrrr-rN) r$r%r]rrr'rxryrOs ""$$$ r/ spearmanrhorus`J  B  $DA  %B  %B ..d SC cmmSZZ 88r1)rRroc\V4#rMtupler$rPs&&r/r`r`uQxr1) default_axisrVrTrUrSrWc\WW#RR7#)a Computes the Theil-Sen estimator for a set of points (x, y). `theilslopes` implements a method for robust linear regression. It computes the slope as the median of all slopes between paired values. Parameters ---------- y : array_like Dependent variable. x : array_like or None, optional Independent variable. If None, use ``arange(len(y))`` instead. alpha : float, optional Confidence degree between 0 and 1. Default is 95% confidence. Note that `alpha` is symmetric around 0.5, i.e. both 0.1 and 0.9 are interpreted as "find the 90% confidence interval". method : {'joint', 'separate'}, optional Method to be used for computing estimate for intercept. Following methods are supported, * 'joint': Uses ``np.median(y - slope * x)`` as intercept. * 'separate': Uses ``np.median(y) - slope * np.median(x)`` as intercept. The default is 'separate'. .. versionadded:: 1.8.0 axis : int or tuple of ints, default: None If an int or tuple of ints, the axis or axes of the input along which to compute the statistic. The statistic of each axis-slice (e.g. row) of the input will appear in a corresponding element of the output. If ``None``, the input will be raveled before computing the statistic. Returns ------- result : ``TheilslopesResult`` instance The return value is an object with the following attributes: slope : float Theil slope. intercept : float Intercept of the Theil line. low_slope : float Lower bound of the confidence interval on `slope`. high_slope : float Upper bound of the confidence interval on `slope`. See Also -------- siegelslopes : a similar technique using repeated medians Notes ----- The implementation of `theilslopes` follows [1]_. The intercept is not defined in [1]_, and here it is defined as ``median(y) - slope*median(x)``, which is given in [3]_. Other definitions of the intercept exist in the literature such as ``median(y - slope*x)`` in [4]_. The approach to compute the intercept can be determined by the parameter ``method``. A confidence interval for the intercept is not given as this question is not addressed in [1]_. For compatibility with older versions of SciPy, the return value acts like a ``namedtuple`` of length 4, with fields ``slope``, ``intercept``, ``low_slope``, and ``high_slope``, so one can continue to write:: slope, intercept, low_slope, high_slope = theilslopes(y, x) References ---------- .. [1] P.K. Sen, "Estimates of the regression coefficient based on Kendall's tau", J. Am. Stat. Assoc., Vol. 63, pp. 1379-1389, 1968. .. [2] H. Theil, "A rank-invariant method of linear and polynomial regression analysis I, II and III", Nederl. Akad. Wetensch., Proc. 53:, pp. 386-392, pp. 521-525, pp. 1397-1412, 1950. .. [3] W.L. Conover, "Practical nonparametric statistics", 2nd ed., John Wiley and Sons, New York, pp. 493. .. [4] https://en.wikipedia.org/wiki/Theil%E2%80%93Sen_estimator Examples -------- >>> import numpy as np >>> from scipy import stats >>> import matplotlib.pyplot as plt >>> x = np.linspace(-5, 5, num=150) >>> y = x + np.random.normal(size=x.size) >>> y[11:15] += 10 # add outliers >>> y[-5:] -= 7 Compute the slope, intercept and 90% confidence interval. For comparison, also compute the least-squares fit with `linregress`: >>> res = stats.theilslopes(y, x, 0.90, method='separate') >>> lsq_res = stats.linregress(x, y) Plot the results. The Theil-Sen regression line is shown in red, with the dashed red lines illustrating the confidence interval of the slope (note that the dashed red lines are not the confidence interval of the regression as the confidence interval of the intercept is not included). The green line shows the least-squares fit for comparison. >>> fig = plt.figure() >>> ax = fig.add_subplot(111) >>> ax.plot(x, y, 'b.') >>> ax.plot(x, res[1] + res[0] * x, 'r-') >>> ax.plot(x, res[1] + res[2] * x, 'r--') >>> ax.plot(x, res[1] + res[3] * x, 'r--') >>> ax.plot(x, lsq_res[1] + lsq_res[0] * x, 'g-') >>> plt.show() theilslopes)r$alpharpfun_robust_slopes)r%r$r~rrs&&&&$r/r}r}sp != QQr1c\V4#rMrwrys&&r/r`r`rzr1c\WVRR7#)a Computes the Siegel estimator for a set of points (x, y). `siegelslopes` implements a method for robust linear regression using repeated medians (see [1]_) to fit a line to the points (x, y). The method is robust to outliers with an asymptotic breakdown point of 50%. Parameters ---------- y : array_like Dependent variable. x : array_like or None, optional Independent variable. If None, use ``arange(len(y))`` instead. method : {'hierarchical', 'separate'} If 'hierarchical', estimate the intercept using the estimated slope ``slope`` (default option). If 'separate', estimate the intercept independent of the estimated slope. See Notes for details. axis : int or tuple of ints, default: None If an int or tuple of ints, the axis or axes of the input along which to compute the statistic. The statistic of each axis-slice (e.g. row) of the input will appear in a corresponding element of the output. If ``None``, the input will be raveled before computing the statistic. Returns ------- result : ``SiegelslopesResult`` instance The return value is an object with the following attributes: slope : float Estimate of the slope of the regression line. intercept : float Estimate of the intercept of the regression line. See Also -------- theilslopes : a similar technique without repeated medians Notes ----- With ``n = len(y)``, compute ``m_j`` as the median of the slopes of the lines from the point ``(x[j], y[j])`` to all other ``n-1`` points. ``slope`` is then the median of all slopes ``m_j``. Two ways are given to estimate the intercept in [1]_ which can be chosen via the parameter ``method``. The hierarchical approach uses the estimated slope ``slope`` and computes ``intercept`` as the median of ``y - slope*x``. The other approach estimates the intercept separately as follows: for each point ``(x[j], y[j])``, compute the intercepts of all the ``n-1`` lines through the remaining points and take the median ``i_j``. ``intercept`` is the median of the ``i_j``. The implementation computes `n` times the median of a vector of size `n` which can be slow for large vectors. There are more efficient algorithms (see [2]_) which are not implemented here. For compatibility with older versions of SciPy, the return value acts like a ``namedtuple`` of length 2, with fields ``slope`` and ``intercept``, so one can continue to write:: slope, intercept = siegelslopes(y, x) References ---------- .. [1] A. Siegel, "Robust Regression Using Repeated Medians", Biometrika, Vol. 69, pp. 242-244, 1982. .. [2] A. Stein and M. Werman, "Finding the repeated median regression line", Proceedings of the Third Annual ACM-SIAM Symposium on Discrete Algorithms, pp. 409-413, 1992. Examples -------- >>> import numpy as np >>> from scipy import stats >>> import matplotlib.pyplot as plt >>> x = np.linspace(-5, 5, num=150) >>> y = x + np.random.normal(size=x.size) >>> y[11:15] += 10 # add outliers >>> y[-5:] -= 7 Compute the slope and intercept. For comparison, also compute the least-squares fit with `linregress`: >>> res = stats.siegelslopes(y, x) >>> lsq_res = stats.linregress(x, y) Plot the results. The Siegel regression line is shown in red. The green line shows the least-squares fit for comparison. >>> fig = plt.figure() >>> ax = fig.add_subplot(111) >>> ax.plot(x, y, 'b.') >>> ax.plot(x, res[1] + res[0] * x, 'r-') >>> ax.plot(x, lsq_res[1] + lsq_res[0] * x, 'g-') >>> plt.show() siegelslopes)r$rrr)r%r$rrs&&&$r/rrsX !n EEr1r~c na!VR8XdRMRpW5R09d\RV RV R24h\W4o!\WRS!R7wrVf/S!PVPR,VP R 7MTpS!P W4wr\WS!R 7wrVPR,^8gVPR,^8d \R 4hVR RS!P3,VR S!PR3,, pVR RS!P3,VR S!PR3,, pVR8XdS!PS!PS!PVPR R 4^R7S!P4p\S!4'dVR V3,VR V3,rvMrS!PWP4pWh,Wx,rvS!PV.VPR R ORN54pS!PV.VPR R ORN54p\!S!4'dVP"^8HMV^8Hp \$P&!V4V ,P)S!P*4pWv, p \!S!4'dV!3Rlp MRp Rp VR8XdV !V !V RR7RR7M V !V RR7p VR!9d0V !W R \,P3,V,, RR7pMpVR8Xd!V !VRR7W!VRR7,, pMIVR RS!P3,WR RS!P3,,, pV !V !VRR7RR7pVR8Xd\/V R",VR",R7#VR8d RV, p\1\2P4!VR, 44p\7VRRR7wpp\7VRRR7wppS!P9S!P;V 4RRR7pS!P=VVP R 7p\?VRRR7pR#VV^, ,^V,^,,S!PAVV^, ,^V,^,,RRR7, S!PAVV^, ,^V,^,,RRR7, ,pS!PCS!PEVS!PGV444pS!PIS!PS!PKVVV,, R, 4S!PL4S!PVS!PL4^, 4pS!PES!PS!PKVVV,,R, 4S!PL4^, S!P=^S!PLR 74pS!PO\$PP!V^R7\$PP!V^R73RR7pS!PSV RR7p S!PUV VRR7pS!PV^8VP4p\$P&!V4V,P)S!P*4pV PV^8Xd V R",MT pVPV^8Xd VR",MTpVR$,pVR%,p VPV^8Xd VR",MTpV PV^8Xd V R",MT p \YVVVV R7#)&r}joint hierarchicalseparatezmethod must be either 'z' or 'separate'. 'z ' is invalid.TrYNr3r\z(`x` and `y` must have length at least 2..:NNN)kc<SPVPVPVPP VP4,R7p\ P !VRVR7#))mask?r)r5r^r_xpisnanrquantile)r$r x_nans_maskedr's&& r/ nanmedian!_robust_slopes..nanmediansFJJqvvAFFQUU[[=P4PJQM>>-4@ @r1c6\P!VRVRR7#)romit)r nan_policyrrr$rs&&r/rrs>>!StG Gr1c4\P!VRVR7#)rrrrs&&r/median_robust_slopes..mediansq#D AAr1rr)slope interceptrg?g@min)r return_ties)rkeepdims)rr)rg)rr low_slope high_sloper>rrrdgqq?).r).r)-rFrrr8rr4rr newaxisastypetriuonesboolr broadcast_toreshaper r^xpxatsetnannprfloatrndtrir count_nonzeroisfiniter5r r!r7maximum zeros_likeminimumroundint64concat atleast_ndr9rrgr)"r%r$r~rr other_methoddeltaxdeltayr;deltax0slopesrrmedslopemedinter interceptszrPnxrepsnyrepsntnysigsqsigmaRuRlRdeltai_nanrrrrr's"&$$$$ @r/rros "m37L J//2<.A#HM34 4  B a4B 7DA12 !''"+QWW -A  q $DA  $DAwwr{Q!''"+/CDDsArzz! "QsBJJ'9%: :F sArzz! "QsBJJ'9%: :F } IIbggbggfll23&78Ag> H B<<#CF^VCF^F<<0A#Y FZZ(;!''#2,(;(;>"1-s~~bq/IJQS TA WWV"W %F   vqr  2E OOEAIu{{ 3E FF5M% $ $RVV ,E$MMQ.HRLHE (  2 If IvJ!*1!4 " )I#-??a#7BZJ 5I'0Z IIr1)rkrur}r) dask.arrayzno take_along_axis)rz+not supported by rankdata (take_along_axis))rlznon-concrete boolean indexing)Ngffffff?r)rlzquantile needs lazy nan_policy)Nr)&numpyrr6scipyrrscipy._externalrrscipy._lib._array_apirrrr r r r scipy.stats._stats_pyr rrrscipy.stats._axis_nan_policyr scipy.stats._stats_mstats_commonrrr__all__r0rCrKrQrkrur}rrrdr1r/rs 2XXX..ABB I8(4  % DEF,TQ*1Q!MW*qW*uW*\W*MGW*t$K=PQK,TQ*1Q!MD9 D9DD9qD9M  D9N D N P+$!$9*?$%'qR$qR 'qRh D O Q,41$9*?$%'eF4eF 'eFPlI$lIr1