o
0iŨ @ s g d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZmZmZmZmZmZmZmZ ddlmZ dd lmZmZ G d
d d eZ d d
Z dd Z dd Z dd Z dd Z!dd Z"dd Z#dNdd Z$d d Z%d d Z&d!d" Z'dOd#d$Z(d%d& Z)d'd( Z*d)d* Z+dPd,d-Z,d.d/ Z-d0d1 Z.d2d3d4d5d6Z/dOd7d8Z0dOd9d:Z1dOd;d<Z2dOd=d>Z3dOd?d@Z4dAdB Z5dCdD Z6edEdFZ7dGdH Z8dIdJ Z9zddKl:m9Z9 W n e;y Y nw G dLdM dMZ z_sum..) getintr type_coercemap _exact_ratiosumitems)
datacountpartialsZ partials_getTtypvaluesr3 r2 totalr+ r+ r, _sum s
rE c C s( z| W S ty t| Y S w r/ )Z is_finiteAttributeErrormathZisfinite)xr+ r+ r, _isfinite s
rI c C s | |u r| S |t u s|tu r| S | t u r|S t|| r |S t| |r%| S t| t r,|S t|t r3| S t| tr?t|tr?|S t| trKt|trK| S d}t|| j|jf )Nz"don't know how to coerce %s and %s)r7 bool
issubclassr float TypeErrorr( )rA Smsgr+ r+ r, r9 s
r9 c C st z| W S ty Y n ttfy | d f Y S w z| j| jfW S ty9 dt| j d}t|w )Nzcan't convert type 'z' to numerator/denominator) as_integer_ratiorF
OverflowError
ValueError numerator denominatorr8 r( rM )rH rO r+ r+ r, r; s
r; c C sh t | |u r | S t|tr| jdkrt}z|| W S ty3 t|tr2|| j|| j Y S w )Nr- )r8 rK r7 rT rL rM r rS )valuerA r+ r+ r, _convert s
rV c C s, t | |}|t| kr| | |kr|S tr/ )r lenrR )arH ir+ r+ r,
_find_lteq s
rZ c C s< t | ||d}|t| d kr | |d |kr |d S t)N)lor- )r rW rR )rX lrH rY r+ r+ r,
_find_rteq" s
r] negative valuec c s( | D ]
}|dk rt ||V qd S )Nr )r )rC errmsgrH r+ r+ r, _fail_neg* s r` c C sJ t | | u r t| } t| }|dk rtdt| \}}}t|| |S )Nr- z%mean requires at least one data point)iterlistrW r rE rV )r> r3 rA rD r? r+ r+ r, r 4 s r c sj zt | W n ty d fdd}t|| }Y nw t| }z| W S ty4 tdd w )Nr c 3 s" t | ddD ]\ }|V qd S )Nr- )start) enumerate)iterablerH r3 r+ r, r? \ s zfmean..countz&fmean requires at least one data point)rW rM r$ ZeroDivisionErrorr )r> r? rD r+ rf r, r N s
r c C s0 z
t ttt| W S ty tdd w )NzGgeometric mean requires a non-empty dataset containing positive numbers)r r r: r# rR r )r> r+ r+ r, r i s r c
C s4 t | | u r t| } d}t| }|dk rtd|dkr;|d u r;| d }t|tjtfr7|dk r5t||S td|d u rGt d|}|}n#t ||u rQt|}t||kr[tdt
dd t ||D \}}}zt | |} t
d d t || D \}}} W n
t
y Y dS w |dkrtd
t|| |S ) Nz.harmonic mean does not support negative valuesr- z.harmonic_mean requires at least one data pointr zunsupported typez*Number of weights does not match data sizec s s | ]}|V qd S r/ r+ )r1 wr+ r+ r, r4 s z harmonic_mean..c s s$ | ]
\}}|r || nd V qdS )r Nr+ )r1 rh rH r+ r+ r, r4 " z Weighted sum must be positive)ra rb rW r
isinstancenumbersZRealr rM r rE r` ziprg rV )
r> Zweightsr_ r3 rH Z sum_weights_rA rD r? r+ r+ r, r | s>
" r c C sZ t | } t| }|dkrtd|d dkr | |d S |d }| |d | | d S Nr no median for empty data r- sortedrW r )r> r3 rY r+ r+ r, r
s r
c C sJ t | } t| }|dkrtd|d dkr | |d S | |d d S rn rq r> r3 r+ r+ r, r
s r
c C s. t | } t| }|dkrtd| |d S )Nr ro rp rq rs r+ r+ r, r s r r- c
C s t | } t| }|dkrtd|dkr| d S | |d }||fD ]}t|ttfr2td| q#z||d }W n tyN t|t|d }Y nw t| |}t | ||}|}|| d } |||d | | S )Nr ro r- rp zexpected number but got %r)
rr rW r rj strbytesrM rL rZ r] )
r> Zintervalr3 rH objLl1l2Zcffr+ r+ r, r s,
r c C s< t t| d}z|d d W S ty tdd w )Nr- r zno mode for empty data)r& ra most_common
IndexErrorr )r> Zpairsr+ r+ r, r , s
r c C sB t t| }tt|tdddg f\}}tttd|S )Nr- )keyr )r& ra r{ nextr r% rb r: )r> ZcountsZmaxcountZ
mode_itemsr+ r+ r, r J s r exclusive)r3 methodc
C s> |dk r t dt| } t| }|dk rt d|dkrM|d }g }td|D ]"}t|| |\}}| | || | |d | | } || q(|S |dkr|d }g }td|D ]9}|| | }|dk rjdn ||d krt|d n|}|| || }| |d || | | | | } || q\|S td|)Nr- zn must be at least 1rp z"must have at least two data pointsZ inclusiver zUnknown method: )r rr rW rangedivmodappendrR )
r> r3 r ZldmresultrY jZdeltaZ interpolatedr+ r+ r, r s4 $ $$ r c s d urt fdd| D \}}}||fS t | \}}}|| \}}t }tt| D ] \}} || | | }
| | } || | |
|
7 < q.d |v rW|d }||fS tdd | D }||fS )Nc 3 | ] }| d V qdS )rp Nr+ )r1 rH cr+ r, r4 z_ss..c s r. r/ r r0 r+ r+ r, r4 r5 )rE rP r& r: r; r< r= ) r> r rA rD r? Zmean_nZmean_dr@ r3 r2 Zdiff_nZdiff_dr+ r r, _ss s r c C sN t | | u r t| } t| }|dk rtdt| |\}}t||d |S )Nrp z*variance requires at least two data pointsr- ra rb rW r r rV )r> xbarr3 rA ssr+ r+ r, r s %r c C sJ t | | u r t| } t| }|dk rtdt| |\}}t|| |S )Nr- z*pvariance requires at least one data pointr )r> mur3 rA r r+ r+ r, r s "r c C 4 t | |}z| W S ty t| Y S w r/ )r r rF rG )r> r varr+ r+ r, r 0
r c C r r/ )r r rF rG )r> r r r+ r+ r, r C r r c sp t | }t ||krtd|dk rtdt| | t|| t fddt| |D }||d S )NzDcovariance requires that both inputs have same number of data pointsrp z,covariance requires at least two data pointsc 3 $ | ]
\}}| | V qd S r/ r+ r1 xiyir ybarr+ r, r4 u ri z covariance..r- )rW r r$ rl )rH yr3 sxyr+ r r, r ] s r c s t | }t ||krtd|dk rtdt| | t|| t fddt| |D }t fdd| D }tfdd|D }z |t|| W S ty\ tdw ) NzEcorrelation requires that both inputs have same number of data pointsrp z-correlation requires at least two data pointsc 3 r r/ r+ r r r+ r, r4 ri z correlation..c 3 r @Nr+ r1 r r r+ r, r4 r c 3 r r r+ )r1 r )r r+ r, r4 r z&at least one of the inputs is constant)rW r r$ rl r rg )rH r r3 r sxxZsyyr+ r r, r y s r LinearRegressionslope interceptc s t | }t ||krtd|dk rtdt| | t|| t fddt| |D }t fdd| D }z|| }W n tyN tdw | }t||dS ) NzKlinear regression requires that both inputs have same number of data pointsrp z3linear regression requires at least two data pointsc 3 r r/ r+ r r r+ r, r4 ri z$linear_regression..c 3 r r r+ r r r+ r, r4 r z
x is constantr )rW r r$ rl rg r )rH r r3 r r r r r+ r r, r s" r c C s | d }t |dkrXd|| }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 sigmaqrZnumZdenrH r+ r+ r, _normal_dist_inv_cdf sd r )r c @ s e Zd Z dddZd=ddZedd 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 d9d: Z d;d< Z!d
S )?r z(Arithmetic mean of a normal distributionz+Standard deviation of a normal distribution_mu_sigmar r c C s* |dk r t dt|| _t|| _d S )Nr zsigma must be non-negative)r rL r r )selfr r r+ r+ r, __init__% s
zNormalDist.__init__c C s0 t |ttfs t|}t|}| |t||S r/ )rj rb tupler r )clsr> r r+ r+ r, from_samples, s
zNormalDist.from_samplesN)seedc sD |d u rt jnt |j | j| j fddt|D S )Nc s g | ]} qS r+ r+ r1 rY gaussr r r+ r,
8 s z&NormalDist.samples..)randomr ZRandomr r r )r r3 r r+ r r, samples4 s zNormalDist.samplesc C s> | j d }|s tdt|| j d d| tt| S )Nr z$pdf() not defined when sigma is zerog )r r r r r r" )r rH r r+ r+ r, pdf: s
&zNormalDist.pdfc C s4 | j stdddt|| j | j td S )Nz$cdf() not defined when sigma is zeror r r )r r r! r r r rH r+ r+ r, cdfA s $zNormalDist.cdfc C s< |dks |dkr
t d| jdkrt 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_cdfG s
zNormalDist.inv_cdfr c s fddt d D S )Nc s g | ] } | qS r+ )r r r3 r r+ r, r ` s z(NormalDist.quantiles..r- )r )r r3 r+ r r, r W s zNormalDist.quantilesc C s t |ts
td| |}}|j|jf|j|jfk r ||}}|j|j}}|r+|s/td|| }t|j|j }|sLdt|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 ) rj r rM r r r r r r! r r# r ) r otherXYZX_varZY_varZdvZdmrX bx1x2r+ r+ r, overlapb s$
( 4zNormalDist.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 | j S r/ r r r+ r+ r, r zNormalDist.meanc C r r/ r r r+ r+ r, r
r zNormalDist.medianc C r r/ r r r+ r+ r, r s zNormalDist.modec C r r/ r r r+ r+ r, r r zNormalDist.stdevc C s | j d S )Nr r r r+ r+ r, r s
zNormalDist.variancec C s: t |trt| j|j t| j|jS t| j| | jS r/ rj r r r r r r r+ r+ r, __add__
zNormalDist.__add__c C s: t |trt| j|j t| j|jS t| j| | jS r/ r r r+ r+ r, __sub__ r zNormalDist.__sub__c C s t | j| | jt| S r/ r r r r r r+ r+ r, __mul__ zNormalDist.__mul__c C s t | j| | jt| S r/ r r r+ r+ r, __truediv__ r zNormalDist.__truediv__c C s t | j| jS r/ r r r r r+ r+ r, __pos__ s zNormalDist.__pos__c C s t | j | jS r/ r r r+ r+ r, __neg__ 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/ )rj r NotImplementedr r r r+ r+ r, __eq__ s
zNormalDist.__eq__c C s t | j| jfS r/ )hashr r r r+ r+ r, __hash__ r zNormalDist.__hash__c C s t | j d| jd| jdS )Nz(mu=z, sigma=))r8 r( r r r r+ r+ r, __repr__ s zNormalDist.__repr__c C s | j | jfS r/ r r r+ r+ r, __getstate__ s zNormalDist.__getstate__c C s |\| _ | _d S r/ r )r stater+ r+ r, __setstate__ s zNormalDist.__setstate__)r r )r )"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+ r, r sN
"
r )r^ r/ )r- )=__all__rG rk r Z fractionsr decimalr itertoolsr r Zbisectr r r r r r r! r" r# r$ operatorr% collectionsr& r' rR r rE rI r9 r; rV rZ r] r` r r r 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` j ( 4
8
7 7
8
/
,
!-K