a
XC?h @ s g d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z ddlm
Z
mZmZmZmZmZmZmZ ddlmZ dd lmZ G d
d d eZdFd d
Zdd Z dd Z dd Z dd Z dd Z dd Z!dGdd Z"d d Z#d d Z$d!d" Z%d#d$ Z&d%d& Z'd'd( Z(d)d* Z)dHd,d-Z*d.d/ Z+d0d1 Z,d2d3d4d5d6Z-dId7d8Z.dJd9d:Z/dKd;d<Z0dLd=d>Z1dMd?d@Z2dAdB Z3zddCl4m3Z3 W n e5y Y n0 G dDdE dEZ6dS )N)
NormalDistStatisticsErrorfmeangeometric_mean
harmonic_meanmeanmedianmedian_grouped median_high
median_lowmode multimodepstdev pvariance quantilesstdevvariance NFraction)Decimal)groupby) bisect_left bisect_right)hypotsqrtfabsexperftaulogfsum)
itemgetter)Counterc @ s e Zd ZdS )r N)__name__
__module__ __qualname__ r& r& //opt/alt/python39/lib64/python3.9/statistics.pyr u s r c C s d}t |\}}||i}|j}ttt|}t| tD ]@\}} t||}tt | D ]"\}}|d7 }||d| ||< qRq6d |v r|d }
ntdd t| D }
||
|fS )Nr c s s | ]\}}t ||V qd S Nr ).0dnr& r& r' z_sum..)
_exact_ratioget_coerceinttyper mapsumsorteditems) datastartcountr, r+ ZpartialsZ partials_getTtypvaluestotalr& r& r' _sum{ s $
r? c C s, z
| W S ty& t| Y S 0 d S r) )Z is_finiteAttributeErrormathZisfinite)xr& r& r' _isfinite s
rC c C s | |u r | S |t u s |tu r | S | t u r,|S t|| r:|S t| |rH| S t| t rV|S t|t rd| S t| tr|t|tr||S t| trt|tr| S d}t|| j|jf d S )Nz"don't know how to coerce %s and %s)r2 bool
issubclassr float TypeErrorr# )r; Smsgr& r& r' r1 s r1 c C s znt | tu st | tu r$| W S z| j| jfW W S tyj z| W Y W S tyd Y n0 Y n0 W n ttfy | d f Y S 0 d}t |
t | j d S )Nz0can't convert type '{}' to numerator/denominator) r3 rF r as_integer_ratio numerator denominatorr@
OverflowError
ValueErrorrG formatr# )rB rI r& r& r' r/ s
r/ c C sn t | |u r| S t|tr(| jdkr(t}z
|| W S tyh t|trb|| j|| j Y S Y n0 d S )Nr( )r3 rE r2 rL rF rG r rK )valuer; r& r& r' _convert s
rQ c C s. t | |}|t| kr&| | |kr&|S td S r) )r lenrN )arB ir& r& r'
_find_lteq
s
rU c C s> t | ||d}|t| d kr6| |d |kr6|d S td S )N)lor( )r rR rN )rS lrB rT r& r& r'
_find_rteq s rX negative valuec c s$ | D ]}|dk rt ||V qd S )Nr )r )r= errmsgrB r& r& r' _fail_neg s r[ c C sH t | | u rt| } t| }|dk r,tdt| \}}}t|| |S )Nr( z%mean requires at least one data point)iterlistrR r r? rQ )r8 r, r; r> r: r& r& r' r ' s r c sp z t | W n. ty: d fdd}t|| }Y n
0 t| }z
| W S tyj tdd Y n0 d S )Nr c 3 s t | ddD ]\ }|V q d S )Nr( )r9 ) enumerate)iterablerB r, r& r' r: O s zfmean..countz&fmean requires at least one data point)rR rG r ZeroDivisionErrorr )r8 r: r> r& r` r' r A s
r c C s6 zt ttt| W S ty0 tdd Y n0 d S )NzGgeometric mean requires a non-empty dataset containing positive numbers)r r r4 r rN r )r8 r& r& r' r \ s r c C s t | | u rt| } d}t| }|dk r2tdn<|dkrn| d }t|tjtfrf|dk rbt||S tdz"t dd t
| |D \}}}W n t y Y dS 0 t || |S )Nz.harmonic mean does not support negative valuesr( z.harmonic_mean requires at least one data pointr zunsupported typec s s | ]}d | V qdS )r( Nr& r* rB r& r& r' r- r. z harmonic_mean..)
r\ r] rR r
isinstancenumbersZRealr rG r? r[ ra rQ )r8 rZ r, rB r; r> r: r& r& r' r o s$
" r c C s\ t | } t| }|dkr td|d dkr8| |d S |d }| |d | | d S d S Nr no median for empty data r( r6 rR r )r8 r, rT r& r& r' r s
r c C sL t | } t| }|dkr td|d dkr8| |d S | |d d S d S re rh r8 r, r& r& r' r
s r
c C s, t | } t| }|dkr td| |d S )Nr rf rg rh ri r& r& r' r s
r r( c
C s t | } t| }|dkr"tdn|dkr2| d S | |d }||fD ] }t|ttfrFtd| qFz||d }W n& ty t|t|d }Y n0 t| |}t | ||}|}|| d } |||d | | S )Nr rf r( rg zexpected number but got %r)
r6 rR r rc strbytesrG rF rU rX )
r8 Zintervalr, rB objLl1l2Zcffr& r& r' r s&
r c C sB t t| d}z|d d W S ty< tdd Y n0 d S )Nr( r zno mode for empty data)r" r\ most_common
IndexErrorr )r8 Zpairsr& r& r' r s
r c C s@ t t| }tt|tdddg f\}}tttd|S )Nr( )keyr )r" r\ rq nextr r! r] r4 )r8 ZcountsZmaxcountZ
mode_itemsr& r& r' r 4 s
r exclusive)r, methodc
C sB |dk rt dt| } t| }|dk r0t d|dkr|d }g }td|D ]D}t|| |\}}| | || | |d | | } || qN|S |dkr0|d }g }td|D ]r}|| | }|dk rdn||d kr|d n|}|| || }| |d || | | | | } || q|S td|d S )Nr( zn must be at least 1rg z"must have at least two data pointsZ inclusiverv zUnknown method: )r r6 rR rangedivmodappendrN )
r8 r, rw ZldmresultrT jZdeltaZ interpolatedr& r& r' r k s2 $
$$ r c s d ur,t fdd| D \}}}||fS t| t fdd| D \}}}t fdd| D \}}}||d t| 8 }||fS )Nc 3 s | ]}| d V qdS rg Nr& rb cr& r' r- r. z_ss..c 3 s | ]}| d V qdS r~ r& rb r r& r' r- r. c 3 s | ]}| V qd S r) r& rb r r& r' r- r. rg )r? r rR )r8 r r; r> r: UZtotal2Zcount2r& r r' _ss s r c C sL t | | u rt| } t| }|dk r,tdt| |\}}t||d |S )Nrg z*variance requires at least two data pointsr( r\ r] rR r r rQ )r8 xbarr, r; ssr& r& r' r s & r c C sH t | | u rt| } t| }|dk r,tdt| |\}}t|| |S )Nr( z*pvariance requires at least one data pointr )r8 mur, r; r r& r& r' r s # r c C s6 t | |}z
| W S ty0 t| Y S 0 d S r) )r r r@ rA )r8 r varr& r& r' r s
r c C s6 t | |}z
| W S ty0 t| Y S 0 d S r) )r r r@ rA )r8 r r r& r& r' r
$ s
r
c C s | d }t |dkrd|| }d| d | d | d | d | d | d
| d | }d | d
| d | d | d | d | d | d }|| }||| S |dkr| nd| }tt| }|dkr^|d }d| d | d | d | d | d | d | d }d | d | d! | d" | d# | d$ | d% | d }n|d }d&| d' | d( | d) | d* | d+ | d, | d- }d.| d/ | d0 | d1 | d2 | d3 | d4 | d }|| }|dk r| }||| S )5N ?g333333?g Q ?g^}o)@gE.kR@g U l@g*u>l@g N@g"]Ξ@gnC`@gu @giK~j@gv|E@gd|1@gfRr@gu.2@g~y@gn8(E@ ? g @g?g鬷ZaI?ggElD?g7\?guSS?g=.
@gj%b@gHw@gjRe?g9dh?
>g('߿A?g~z ?g@3?gɅ3?g3fRx?gIFl @gt>g*Yn>gESB\T?gN;A+?gUR1?gE F?gPn@g&>@gigtcI,\>gŝI?g*F2v?gC4?gO1?)r r r )pr sigmaqrZnumZdenrB r& r& r' _normal_dist_inv_cdf7 sd
r )r c @ s e Zd ZdddZd9ddZedd Zd
d d d
Zdd Zdd Z dd Z
d:ddZ dd Z dd Z
edd Zed d Zed d Zed!d" Zed#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 ZeZd1d2 ZeZ d3d4 Z d5d6 Z d7d8 Z d
S );r z(Arithmetic mean of a normal distributionz+Standard deviation of a normal distribution)_mu_sigmar r c C s( |dk rt dt|| _t|| _d S )Nr zsigma must be non-negative)r rF r r )selfr r r& r& r' __init__ s
zNormalDist.__init__c C s. t |ttfst|}t|}| |t||S r) )rc r] tupler r )clsr8 r r& r& r' from_samples s zNormalDist.from_samplesN)seedc sB |d u rt jn
t |j | j| j fddt|D S )Nc s g | ]} qS r& r& r* rT gaussr r r& r'
r. z&NormalDist.samples..)randomr ZRandomr r rx )r r, r r& r r' samples s zNormalDist.samplesc C s< | j d }|stdt|| j d d| tt| S )N @z$pdf() not defined when sigma is zerog )r r r r r r )r rB r r& r& r' pdf s
zNormalDist.pdfc C s2 | j stdddt|| j | j td S )Nz$cdf() not defined when sigma is zeror r r )r r r r r r rB r& r& r' cdf s zNormalDist.cdfc C s: |dks|dkrt d| jdkr*t dt|| j| jS )Nr r z$p must be in the range 0.0 < p < 1.0z-cdf() not defined when sigma at or below zero)r r r r )r r r& r& r' inv_cdf s
zNormalDist.inv_cdfru c s fddt d D S )Nc s g | ]} | qS r& )r r r, r r& r' r r. z(NormalDist.quantiles..r( )rx )r r, r& r r' r s zNormalDist.quantilesc C s t |tstd| | }}|j|jf|j|jfk r>|| }}|j|j }}|rT|s\td|| }t|j|j }|sdt|d|j t d S |j| |j| }|j|j t |d |t
|| } || | }
|| | } dt| |
| |
t| | | | S )Nz$Expected another NormalDist instancez(overlap() not defined when sigma is zeror r ) rc r rG r r r r r r r r r ) r otherXYZX_varZY_varZdvZdmrS bx1x2r& r& r' overlap s"
( zNormalDist.overlapc C s | j std|| j | j S )Nz'zscore() not defined when sigma is zero)r r r r r& r& r' zscore s zNormalDist.zscorec C s | j S r) r r r& r& r' r s zNormalDist.meanc C s | j S r) r r r& r& r' r s zNormalDist.medianc C s | j S r) r r r& r& r' r s zNormalDist.modec C s | j S r) r r r& r& r' r s zNormalDist.stdevc C s
| j d S )Nr r r r& r& r' r s zNormalDist.variancec C s8 t |tr&t| j|j t| j|jS t| j| | jS r) rc r r r r r r r& r& r' __add__ s
zNormalDist.__add__c C s8 t |tr&t| j|j t| j|jS t| j| | jS r) r r r& r& r' __sub__' s
zNormalDist.__sub__c C s t | j| | jt| S r) r r r r r r& r& r' __mul__5 s zNormalDist.__mul__c C s t | j| | jt| S r) r r r& r& r' __truediv__= s zNormalDist.__truediv__c C s t | j| jS r) r r r r r& r& r' __pos__E s zNormalDist.__pos__c C s t | j | jS r) r r r& r& r' __neg__I s zNormalDist.__neg__c C s
| | S r) r& r r& r& r' __rsub__O s zNormalDist.__rsub__c C s& t |tstS | j|jko$| j|jkS r) )rc r NotImplementedr r r r& r& r' __eq__U s
zNormalDist.__eq__c C s t | j| jfS r) )hashr r r r& r& r' __hash__[ s zNormalDist.__hash__c C s t | j d| jd| jdS )Nz(mu=z, sigma=))r3 r# r r r r& r& r' __repr___ s zNormalDist.__repr__)r r )ru ) r# r$ r% __slots__r classmethodr r r r r r r r propertyr r r r r r r r r r r __radd__r __rmul__r r r r& r& r& r' r sF
"
r )r )rY )r( )N)N)N)N)N)7__all__rA rd r Z fractionsr Zdecimalr itertoolsr Zbisectr r r r r r r r r r operatorr! collectionsr" rN r r? rC r1 r/ rQ rU rX r[ r r r r r r
r r r r r r r r r r
r Z _statistics ImportErrorr r& r& r& r' S sR (
:
/
7 78
/
,
K