U
e5d
@ sj d ddddddddd d
d d d
dddgZ ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z ddlm
Z
mZmZmZmZmZmZmZ ddlmZ ddlmZ G dd deZdcdd Zd d Z d d Z d!d" Z d#d$ Z d%d& Z d'd( Z!ddd*d+Z"d,d Z#d-d Z$d.d Z%d/d Z&d0d Z'd1d Z(d2d Z)ded4dZ*d5d
Z+d6d Z,d7d8d9d:dZ-dfd;d<Z.dgd=dZ/dhd>d
Z0did?dZ1djd@d Z2dAdB Z3G dCd d Z4zddDl5m3Z3 W n e6k
r Y nX e7dEkrfddFlm8Z8 ddGlm9Z9m:Z:m;Z;mZ>e4dIdJZ?e4dKdLZ@dMZAe?BeAZCe@BeAZDe9e:fD ]P dS )k
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& "/usr/lib64/python3.8/statistics.pyr u s c C s d}t |\}}||i}|j}ttt|}t| tD ]@\}} t||}tt | D ]"\}}|d7 }||d| ||< qRq6d |kr|d }
ntdd t| D }
||
|fS )Nr c s s | ]\}}t ||V qd S Nr ).0dnr& r& r' s z_sum..)
_exact_ratioget_coerceinttyper mapsumsorteditems) datastartcountr, r+ ZpartialsZ partials_getTtypvaluestotalr& r& r' _sum{ s $
r> c C s. z
| W S tk
r( t| Y S X d S r) )Z is_finiteAttributeErrormathZisfinite)xr& r& r' _isfinite s
rB c C s | |kr | S |t ks |tkr | S | t kr,|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)r1 bool
issubclassr float TypeErrorr# )r: Smsgr& r& r' r0 s(
r0 c C s zrt | tkst | tkr$| W S z| j| jfW W S tk
rn z| W Y W S tk
rh Y nX Y nX W n ttfk
r | d f Y S X d}t |
t | j d S )Nz0can't convert type '{}' to numerator/denominator) r2 rE r as_integer_ratio numerator denominatorr?
OverflowError
ValueErrorrF formatr# )rA rH r& r& r' r. s
r. c C sp t | |kr| S t|tr(| jdkr(t}z
|| W S tk
rj t|trd|| j|| j Y S Y nX d S )Nr( )r2 rD r1 rK rE rF r rJ )valuer: r& r& r' _convert s
rP c C s. t | |}|t| kr&| | |kr&|S td S r) )r lenrM )arA ir& r& r'
_find_lteq
s
rT c C s> t | ||d}|t| d kr6| |d |kr6|d S td S )N)lor( )r rQ rM )rR lrA rS r& r& r'
_find_rteq s rW negative valuec c s$ | D ]}|dk rt ||V qd S )Nr )r )r< errmsgrA r& r& r' _fail_neg s rZ c C sH t | | krt| } t| }|dk r,tdt| \}}}t|| |S )Nr( z%mean requires at least one data point)iterlistrQ r r> rP )r7 r, r: r= r9 r& r& r' r ' s c st z t | W n0 tk
r< d fdd}t|| }Y n
X t| }z
| W S tk
rn tdd Y nX d S )Nr c 3 s t | ddD ]\ }|V q d S )Nr( )r8 ) enumerate)iterablerA r, r& r' r9 O s zfmean..countz&fmean requires at least one data point)rQ rF r ZeroDivisionErrorr )r7 r9 r= r& r_ r' r A s
c C s8 zt ttt| W S tk
r2 tdd Y nX d S )NzHgeometric mean requires a non-empty dataset containing positive numbers)r r r3 r rM r )r7 r& r& r' r \ s c C s t | | krt| } d}t| }|dk r2tdn<|dkrn| d }t|tjtfrf|dk rbt||S tdz"t dd t
| |D \}}}W n t k
r Y dS X 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* rA r& r& r' r- s z harmonic_mean..)
r[ r\ rQ r
isinstancenumbersZRealr rF r> rZ r` rP )r7 rY r, rA r: r= r9 r& r& r' r o s$
"c C s\ t | } t| }|dkr td|d dkr8| |d S |d }| |d | | d S d S Nr no median for empty data r( r5 rQ r )r7 r, rS r& r& r' r s
c C sL t | } t| }|dkr td|d dkr8| |d S | |d d S d S rd rg r7 r, r& r& r' r
s c C s, t | } t| }|dkr td| |d S )Nr re rf rg rh r& r& r' r s
r( c
C s t | } t| }|dkr"tdn|dkr2| d S | |d }||fD ] }t|ttfrFtd| qFz||d }W n( tk
r t|t|d }Y nX t| |}t | ||}|}|| d } |||d | | S )Nr re r( rf zexpected number but got %r)
r5 rQ r rb strbytesrF rE rT rW )
r7 Zintervalr, rA objLl1l2Zcffr& r& r' r s&
c C sH t | } t| d}z|d d W S tk
rB tdd Y nX d S )Nr( r zno mode for empty data)r[ r" most_common
IndexErrorr )r7 Zpairsr& r& r' r s c C s@ t t| }tt|tdddg f\}}tttd|S )Nr( )keyr )r" r[ rp nextr r! r\ r3 )r7 ZcountsZmaxcountZ
mode_itemsr& r& r' r 5 s
exclusive)r, methodc
C sL |dk rt dt| } t| }|dk r0t d|dkr|d }g }td|D ]N}|| | }|| || }| | || | |d | | } || qN|S |dkr:|d }g }td|D ]r}|| | }|dk rdn||d kr|d n|}|| || }| |d || | | | | } || q|S td|d S )Nr( zn must be at least 1rf z"must have at least two data pointsZ inclusiveru zUnknown method: )r r5 rQ rangeappendrM )
r7 r, rv ZldmresultrS jZdeltaZ interpolatedr& r& r' r l s4 $
$$ c s d k r,t fdd| D \}}}||fS t| t fdd| D \}}}t fdd| D \}}}||d t| 8 }||fS )Nc 3 s | ]}| d V qdS rf Nr& ra cr& r' r- s z_ss..c 3 s | ]}| d V qdS r| r& ra r} r& r' r- s c 3 s | ]}| V qd S r) r& ra r} r& r' r- s rf )r> r rQ )r7 r~ r: r= r9 UZtotal2Zcount2r& r} r' _ss s r c C sL t | | krt| } t| }|dk r,tdt| |\}}t||d |S )Nrf z*variance requires at least two data pointsr( r[ r\ rQ r r rP )r7 xbarr, r: ssr& r& r' r s & c C sH t | | krt| } t| }|dk r,tdt| |\}}t|| |S )Nr( z*pvariance requires at least one data pointr )r7 mur, r: r r& r& r' r s # c C s8 t | |}z
| W S tk
r2 t| Y S X d S r) )r r r? r@ )r7 r varr& r& r' r s
c C s8 t | |}z
| W S tk
r2 t| Y S X d S r) )r r r? r@ )r7 r r r& r& r' r
& s
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 sigmaqrZnumZdenrA r& r& r' _normal_dist_inv_cdf9 sd
r c @ s e Zd ZdddZd7ddZedd Zd
d d d
Zdd Zdd Z dd Z
d8ddZ dd Z e
dd Ze
dd Ze
d d Ze
d d Ze
d!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. ZeZd/d0 ZeZd1d2 Z d3d4 Z d5d6 Z d
S )9r 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 rE r r )selfr r r& r& r' __init__ s
zNormalDist.__init__c C s. t |ttfst|}t|}| |t||S r) )rb r\ tupler r )clsr7 r r& r& r' from_samples s zNormalDist.from_samplesN)seedc sB |d krt jn
t |j | j| j fddt|D S )Nc s g | ]} qS r& r& r* rS gaussr r r& r'
s z&NormalDist.samples..)randomr ZRandomr r rw )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 rA 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 rA 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_cdfrt c s fddt d D S )Nc s g | ]} | qS r& )r r r, r r& r' r s z(NormalDist.quantiles..r( )rw )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 ) rb r rF r r r r r r r r r ) r otherXYZX_varZY_varZdvZdmrR bx1x2r& r& r' overlap s"
( zNormalDist.overlapc 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) rb 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__% 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__5 s zNormalDist.__pos__c C s t | j | jS r) r r r& r& r' __neg__9 s zNormalDist.__neg__c C s
| | S r) r& r r& r& r' __rsub__? s zNormalDist.__rsub__c C s& t |tstS | j|jko$| j|jkS r) )rb r NotImplementedr r r r& r& r' __eq__E s
zNormalDist.__eq__c C s t | j| jfS r) )hashr r r r& r& r' __hash__K s zNormalDist.__hash__c C s t | j d| jd| jdS )Nz(mu=z, sigma=))r2 r# r r r r& r& r' __repr__O s zNormalDist.__repr__)r r )rt ) r# r$ r% __slots__r classmethodr 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 sD
"
)r __main__)isclose)addsubmultruediv)repeat
i z
Test z with another NormalDist: z with a constant: z
Test constant with :c C s d S r) r& )G1G2r& r& r' assert_close s r iI / g `@@c C s g | ] }|t qS r& sra r& r& r' r s r c C s g | ] }|t qS r& r ra r& r& r' r s c C s g | ] }|t qS r& r ra r& r& r' r s c C s g | ] }|t qS r& r ra r& r& r' r s c C s g | ]\}}|| qS r& r& r* rA yr& r& r' r s c C s g | ]\}}|| qS r& r& r r& r& r' r s )r )rX )r( )N)N)N)N)N)Q__all__r@ rc r Z fractionsr Zdecimalr itertoolsr Zbisectr r r r r r r r r r operatorr! collectionsr" rM r r> rB r0 r. rP rT rW rZ 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 r r ZdoctestZg1Zg2r, r r r funcprintr r3 Zconstr r r r rG zipZtestmodr& r& r& r' T s (
:
/
7 79
/
,
J Q