B
    Z                 @   s   d Z ddlZddlmZ ddlZddlmZ dddZe	dkrddl
mZ ddlZejj Zejejdd	e_eejZeejdddd
f ZejdredZeejZeeejejfZeedd ejjD Z dS )z,
Glue for returning descriptive statistics.
    N)stats)	sign_testc                s0  t |  |dkr8t t js8 jdkr8 dddf   jd dkrhd|dt t d d t d d  	 t 
 dt   d t   d     t t t t j d	 dd
t j  	  d	 dd
t  d }|dt fdddD  7 }t d\}}t \}}tt  \}}	|d||||||	f 7 }nĈ jd dkr$dtj }xt jd D ]}
|d|
t dd|
f  dd|
f 	  dd|
f  dt dd|
f   d t dd|
f   d tj d 7 }qW ntd|S )a  
    Prints descriptive statistics for one or multiple variables.

    Parameters
    ------------
    data: numpy array
        `x` is the data

    v: list, optional
        A list of the column number or field names (for a recarray) of variables.
        Default is all columns.

    axis: 1 or 0
        axis order of data.  Default is 0 for column-ordered data.

    Examples
    --------
    >>> descstats(data.exog,v=['x_1','x_2','x_3'])
    N   a  
    ---------------------------------------------
    Univariate Descriptive Statistics
    ---------------------------------------------

    Var. Name   %(name)12s
    ----------
    Obs.          %(nobs)22i  Range                  %(range)22s
    Sum of Wts.   %(sum)22s  Coeff. of Variation     %(coeffvar)22.4g
    Mode          %(mode)22.4g  Skewness                %(skewness)22.4g
    Repeats       %(nmode)22i  Kurtosis                %(kurtosis)22.4g
    Mean          %(mean)22.4g  Uncorrected SS          %(uss)22.4g
    Median        %(median)22.4g  Corrected SS            %(ss)22.4g
    Variance      %(variance)22.4g  Sum Observations        %(sobs)22.4g
    Std. Dev.     %(stddev)22.4g
    zN/Ar   (z, )   )axis)namesumZnobsmodeZnmodemeanmedianrangeZvariancestddevZcoeffvarZskewnesskurtosisZussZssZsobsa  

    Percentiles
    -------------
    1  %%          %12.4g
    5  %%          %12.4g
    10 %%          %12.4g
    25 %%          %12.4g

    50 %%          %12.4g

    75 %%          %12.4g
    90 %%          %12.4g
    95 %%          %12.4g
    99 %%          %12.4g
    c                s   g | ]}t  |qS  )r   Zscoreatpercentile).0Zper)xr   <lib/python3.7/site-packages/statsmodels/sandbox/descstats.py
<listcomp>T   s    zdescstats.<locals>.<listcomp>)	r      
      2   K   Z   _   c   ac  

    Tests of Location (H0: Mu0=0)
    -----------------------------
    Test                Statistic       Two-tailed probability
    -----------------+-----------------------------------------
    Student's t      |  t %7.5f   Pr > |t|   <%.4f
    Sign             |  M %8.2f   Pr >= |M|  <%.4f
    Signed Rank      |  S %8.2f   Pr >= |S|  <%.4f

    z
    Var. Name   |     Obs.        Mean    Std. Dev.           Range
    ------------+--------------------------------------------------------z;%(name)15s %(obs)9i %(mean)12.4g %(stddev)12.4g %(range)20s)r	   Zobsr   r   r   zdata not understood)npZarray
isinstanceZrecarrayndimshapelenr   r   r   r   strminmaxvarZstdZ	variationZskewr   r
   tupleZttest_1sampr   ZwilcoxonZsqueezeoslinesepr   
ValueError)dataZcolsr   ZdesctZp_tMZp_MSZp_Sr&   r   )r   r   	descstats   s<    
 "(

(Rr/   __main__F)Zprependr   z./Econ724_PS_I_Data.csvc             C   s   g | ]}t | qS r   )data2)r   _r   r   r   r      s   r   )Nr   )!__doc__Znumpyr   Zscipyr   r(   Z"statsmodels.stats.descriptivestatsr   r/   __name__Zstatsmodels.apiZapiZsmZdatasetsZlongleyloadr+   Zadd_constantZexogZsum1Zsum1apathisfileZ
recfromcsvr1   ZaheZsum2Zcolumn_stackZyrseducZsum3ZdtypenamesZsum4r   r   r   r   <module>   s&   	
 


