
    (ph%-                         S SK r S SKrSSKJr  SSKJr  SSKJr	  / SQr
\" S/ S	Q5      r\" S
SS/5      rSS jrS rSS jrg)    N   )distributions   )_make_tuple_bunch)siegelslopes)_find_repeatstheilslopesr   TheilslopesResultslope	intercept	low_slope
high_slopeSiegelslopesResultr   r   c           	      8   US;  a  [        SU S35      e[        R                  " U [        SS9R	                  5       n Uc#  [        R
                  " [        U 5      [        S9nOd[        R                  " U[        SS9R	                  5       n[        U5      [        U 5      :w  a$  [        S[        U 5       S	[        U5       S
35      eUSS2[        R                  4   U-
  nU SS2[        R                  4   U -
  nXTS:     XDS:     -  nUR                  (       d  Sn[        R                  " U[        SS9  UR                  5         [        R                  " U5      nUS:X  a  [        R                  " XU-  -
  5      n	O0[        R                  " U 5      U[        R                  " U5      -  -
  n	US:  a  SU-
  n[        R                  R!                  US-  5      n
[#        U5      u  p[#        U 5      u  p[        U5      n[        U 5      nSXS-
  -  SU-  S-   -  [%        S U 5       5      -
  [%        S U 5       5      -
  -  n [        R&                  " U5      n[)        [+        [        R,                  " XU-  -
  S-  5      5      [        U5      S-
  5      n[/        [+        [        R,                  " XU-  -   S-  5      5      S-
  S5      nUUU/   n[5        XUS   US   S9$ ! [         [0        4 a#    [        R2                  [        R2                  4n NCf = f)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

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()

)jointseparatez-method must be either 'joint' or 'separate'.'z' is invalid.T)dtypecopyNr   Incompatible lengths ! (<>)r   z"All `x` coordinates are identical.r   )
stacklevelr   g      ?g      ?g       @gqq?r      c              3   B   #    U  H  oUS -
  -  SU-  S-   -  v   M     g7fr   r   r   N .0ks     S/var/www/html/venv/lib/python3.13/site-packages/scipy/stats/_stats_mstats_common.py	<genexpr>theilslopes.<locals>.<genexpr>   #     ?1acacAg.   c              3   B   #    U  H  oUS -
  -  SU-  S-   -  v   M     g7fr   r   r   s     r"   r#   r$      r%   r&   r   )
ValueErrornparrayfloatravelarangelennewaxissizewarningswarnRuntimeWarningsortmedianr   normppfr   sumsqrtminintroundmax
IndexErrornanr
   )yxalphamethoddeltaxdeltayslopesmsgmedslopemedinterz_nxrepsnyrepsntnysigsqsigmaRuRldeltas                        r"   r	   r	      s   V ** #HM3 4 	4 	%d+113AyIIc!fE*HHQe$/557q6SV7Axr#a&KLL q"**}!Fq"**}!FQJ&!"44F;;2c>a8
KKMyy H99QA-.99Q<(RYYq\"99s{U
urz*Aa IAa IA	VB	QBRa4[AbDF+???@???@ AE!RXXreG|R/013v;q=ARXXreG|R/01A5q9Bx  8',QxE!HF F 
# ! !s   BK& &0LLc                 >   [        U 5      S:X  aJ  [        R                  " S[        R                  5      [        R                  " S[        R                  5      4$ [        R
                  " U [        R                  5      R                  5       n U R                  5         [        R                  " S/U SS  U S S :g  45      nX   n[        R                  " [        R                  " U5      U R                  /4-   5      n[        R                  " U5      nUS:  nX%   XE   4$ )Nr   Tr   )r.   r)   r*   float64intpasarrayr,   r4   concatenatenonzeror0   diff)arrchangeunique
change_idxfreqatleast2s         r"   r   r      s    
3x1}xx2::&BGG(<<< **S"**
%
+
+
-CHHJ ^^dVSWCR%89:F[F

6 2sxxj] BCJ77:DaxHT^++    c                 8   US;  a  [        S5      e[        R                  " U 5      R                  5       n Uc#  [        R                  " [        U 5      [        S9nOc[        R                  " U[        S9R                  5       n[        U5      [        U 5      :w  a$  [        S[        U 5       S[        U5       S35      e[        R                  " X[        R                  5      nU R                  U5      UR                  U5      p[        XU5      u  pE[        XES9$ )aJ  
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.

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 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()

)hierarchicalr   z/method can only be 'hierarchical' or 'separate'r   r   r   r   )r   r   )r(   r)   rY   r,   r-   r.   r+   result_typefloat32astypesiegelslopes_pythranr   )r@   rA   rC   r   rH   rI   s         r"   r   r      s    @ 11JKK


1AyIIc!fE*JJq&,,.q6SV7Axr#a&KLLNN1,E88E?AHHUOq-aF;HHAArc   )Ngffffff?r   )Nre   )r1   numpyr)    r   _lib._bunchr   _stats_pythranr   ri   __all__r
   r   r	   r   r   rc   r"   <module>ro      s\       + @
: &&9'BC  '';(/'=? \F~,&lBrc   