U
    Ãmœdî  ã                   @   s:  d Z ddlZddlZej d¡ dZdZdZdZ	dD �]þZ
e dd	e
¡Zd
D �]äZdD �]ØZe dde¡Zg Zg Zg Zg ZeeƒD ]ÒZej eed¡Zejjeefd�Ze ee
f¡Ze dde¡Zee
ƒD ]Zee edd…ef< qÌejje
d�e Ze ee¡e ee¡ Z e !e eejjed�  ¡ e !e¡ e !e¡ e !ee "e¡ ¡ q‚e #e¡Zej#edd�Zej#edd�Zej#edd�Zej#edd…df edd…df eefdd�Z$dgdd„ eeƒD ƒ dd„ ee
ƒD ƒ Z%d &e%¡Z%ej' (ej' )e*¡¡Z+ej' &e+de	 ¡Z,ej-e,e$de%ddd� e	d7 Z	qVqLq4dS )zm
Generate test data sets for lme.

After running this script, run lme_results.R with R
to update the output.
é    NiKQ éd   é   é   )r   é   g      à¿g      ø?)r   r   é   )g      à?r   éÿÿÿÿ)Úsize)Zaxiszgroups,endogc                 C   s   g | ]}d | ‘qS )z
exog_fe_%d© ©Ú.0Úkr	   r	   új/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/regression/tests/results/generate_lme.pyÚ
<listcomp>B   s     r   c                 C   s   g | ]}d | ‘qS )z
exog_re_%dr	   r
   r	   r	   r   r   C   s     ú,zlme%02d.csvz%.3fÚ )ÚfmtÚheaderÚ	delimiterÚcomments).Ú__doc__ÚnumpyÚnpÚosÚrandomÚseedZngroupZn_minZn_maxZdsixÚprZlinspaceZre_sdÚpfÚsigÚparamsZendogZexog_feZexog_reÚgroupsÚrangeÚiÚrandintÚnÚnormalZx_feZzerosZx_reÚuÚjÚreÚdotZexpvalÚappendZonesZconcatenateÚdatar   ÚjoinÚpathÚdirnameÚabspathÚ__file__Zcur_dirÚfnameZsavetxtr	   r	   r	   r   Ú<module>   sh   





  ÿÿÿþ
 ÿ