U
    Ãmœd1  ã                   @   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rÞd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¡rÞe 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 }�qœ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 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   úV/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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*   Ú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&   	
 


ÿ