U
    Kvf1                     @   s   d Z ddlZddlZddlmZ ddlmZ dddZe	dkrddl
m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                    s  t |  |dkr" dddf  |dkrD jdkrD dddf   jd dkrtd|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 }t jd D ]d}
 dd|
f }|
t|| | dt	|
  d t	|  d tj d}|d| 7 }qn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 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uss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   A/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/sandbox/descstats.py
<listcomp>U   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
    ------------+--------------------------------------------------------)r	   Zobsr   r   r   z;%(name)15s %(obs)9i %(mean)12.4g %(stddev)12.4g %(range)20szdata not understood)nparrayndimshapelenr   r   r   r   strminmaxvarZstdZ	variationZskewr   r
   tupleZttest_1sampr   ZwilcoxonZsqueezeoslinesepr   
ValueError)datacolsr   desctZp_tMZp_MSZp_Sr(   Zxvkwargsr   r   r   	descstats   sb    

  "    
(r4   __main__F)prependr   z./Econ724_PS_I_Data.csvc                 C   s   g | ]}t | qS r   )data2)r   _r   r   r   r      s   r   )Nr   )!__doc__r*   Znumpyr    Zscipyr   Z"statsmodels.stats.descriptivestatsr   r4   __name__Zstatsmodels.apiapismZdatasetsZlongleyloadr-   Zadd_constantZexogZsum1Zsum1apathisfileZ
recfromcsvr7   ZaheZsum2Zcolumn_stackZyrseducZsum3ZdtypenamesZsum4r   r   r   r   <module>   s&   	
 


