3
\�P � @ s@ d ddddddddd d
d g Z d d
lZd d
lZd d
lZd d
lZd dlmZ d dlmZ d dlm Z m
Z
d dl m Z m
Z
G dd � d e�Zd4dd�Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd5d$d%�Zd&d � Zd'd � Zd(d� Zd)d� Z d*d� Z d6d,d�Z d-d
� Z d7d.d/�Z d8d0d�Z!d9d1d�Z"d:d2d�Z#d;d3d�Z$d
S )<�StatisticsError�pstdev� pvariance�stdev�variance�median�
median_low� median_high�median_grouped�mean�mode�
harmonic_mean� N)�Fraction)�Decimal)�groupby�chain)� bisect_left� bisect_rightc @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r �"/usr/lib64/python3.6/statistics.pyr c s c C s� d}t |�\}}||i}|j}ttt|��}xRt| t�D ]D\}} t||�}x0tt | �D ]"\}}|d7 }||d�| ||< qVW q8W d |kr�|d }
ntdd� t|j � �D ��}
||
|fS )Nr
� c s s | ]\}}t ||�V qd S )N)r )�.0�d�nr r r � � s z_sum..)
� _exact_ratio�get�_coerce�int�typer �map�sum�sorted�items) �data�start�countr r ZpartialsZ partials_get�T�typ�values�totalr r r �_sumi s $
r. c C s( y| j � S tk
r" tj| �S X d S )N)Z is_finite�AttributeError�mathZisfinite)�xr r r � _isfinite� s r2 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| t�r|t|t�r||S t| t�r�t|t�r�| S d}t|| j|jf ��d S )Nz"don't know how to coerce %s and %s)r! �bool�
issubclassr �float� TypeErrorr )r* �S�msgr r r r � s(
r c C s� ydt | �tkst | �tkr"| j� S y | j| jfS tk
r` y| j� S tk
rZ Y nX Y nX W n ttfk
r� | d fS X d}t |j
t | �j ���d S )Nz0can't convert type '{}' to numerator/denominator) r" r5 r �as_integer_ratio� numerator� denominatorr/ �
OverflowError�
ValueErrorr6 �formatr )r1 r8 r r r r � s
r c C sj t | �|kr| S t|t�r(| jdkr(t}y|| �S tk
rd t|t�r^|| j�|| j� S � Y nX d S )Nr )r" r4 r! r; r5 r6 r r: )�valuer* r r r �_convert� s
r@ c C s` t jt| ��j� }|s|S |d d }x4tdt|��D ]"}|| d |kr6|d |� }P q6W |S )Nr
r )� collections�Counter�iter� most_common�range�len)r' �tableZmaxfreq�ir r r �_counts� s rI c C s. t | |�}|t| �kr&| | |kr&|S t�d S )N)r rF r= )�ar1 rH r r r �
_find_lteq s
rK c C s> t | ||d�}|t| �d kr6| |d |kr6|d S t�d S )N)�lor )r rF r= )rJ �lr1 rH r r r �
_find_rteq s rN �negative valuec c s( x"| D ]}|dk rt |��|V qW d S )Nr
)r )r, �errmsgr1 r r r � _fail_neg s
rQ c C sH t | �| krt| �} t| �}|dk r,td��t| �\}}}t|| |�S )Nr z%mean requires at least one data point)rC �listrF r r. r@ )r' r r* r- r) r r r r
# s c C s� t | �| krt| �} d}t| �}|dk r2td��n<|dkrn| d }t|tjtf�rf|dk rbt|��|S td��y"t dd� t
| |�D ��\}}}W n t k
r� 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 r1 r r r r d s z harmonic_mean..)
rC rR rF r �
isinstance�numbersZRealr r6 r. rQ �ZeroDivisionErrorr@ )r' rP r r1 r* r- r) r r r r = s$
"c C s\ t | �} t| �}|dkr td��|d dkr8| |d S |d }| |d | | d S d S )Nr
zno median for empty data� r )r% rF r )r' r rH r r r r l s
c C sL t | �} t| �}|dkr td��|d dkr8| |d S | |d d S d S )Nr
zno median for empty datarV r )r% rF r )r' r r r r r � s c C s, t | �} t| �}|dkr td��| |d S )Nr
zno median for empty datarV )r% rF r )r' r r r r r � s
r c
C s� t | �} t| �}|dkr"td��n|dkr2| d S | |d }x*||fD ] }t|ttf�rHtd| ��qHW y||d }W n( tk
r� t|�t|�d }Y nX t| |�}t | ||�}|}|| d } |||d | | S )Nr
zno median for empty datar rV zexpected number but got %r)
r% rF r rS �str�bytesr6 r5 rK rN )
r' Zintervalr r1 �obj�L�l1�l2Zcf�fr r r r � s&
c C sB t | �}t|�dkr |d d S |r6tdt|� ��ntd��d S )Nr r
z.no unique mode; found %d equally common valueszno mode for empty data)rI rF r )r' rG r r r r � s c sd � d krt | �� t� fdd�| D ��\}}}t� fdd�| D ��\}}}||d t| � 8 }||fS )Nc 3 s | ]}|� d V qdS )rV Nr )r r1 )�cr r r s z_ss..c 3 s | ]}|� V qd S )Nr )r r1 )r^ r r r s rV )r
r. rF )r' r^ r* r- r) �UZtotal2Zcount2r )r^ r �_ss s r` c C sL t | �| krt| �} t| �}|dk r,td��t| |�\}}t||d |�S )NrV z*variance requires at least two data pointsr )rC rR rF r r` r@ )r' �xbarr r* �ssr r r r " s & c C sH t | �| krt| �} t| �}|dk r,td��t| |�\}}t|| |�S )Nr z*pvariance requires at least one data point)rC rR rF r r` r@ )r' �mur r* rb r r r r Q s ' c C s2 t | |�}y|j� S tk
r, tj|�S X d S )N)r �sqrtr/ r0 )r' ra �varr r r r � s
c C s2 t | |�}y|j� S tk
r, tj|�S X d S )N)r rd r/ r0 )r' rc re r r r r � s
)r
)rO )r )N)N)N)N)N)%�__all__rA �decimalr0 rT Z fractionsr r � itertoolsr r Zbisectr r r= r r. r2 r r r@ rI rK rN rQ r
r r r r r r r` r r r r r r r r �O sB
:
/
7*
/
0