B Z@s`dZdZdZdZdZdZdZdZdd lZ dd l m Z dd l mZmZd d ZddZddZd S)z"Heart Transplant Data, Miller 1976Zrestructuredtextz???zTransplant Survival Dataz_ Miller, R. (1976). Least squares regression with censored dara. Biometrica, 63 (3). 449-464. z1Survival times after receiving a heart transplantzThis data contains the survival time after receiving a heart transplant, the age of the patient and whether or not the survival time was censored. a:: Number of Observations - 69 Number of Variables - 3 Variable name definitions:: death - Days after surgery until death age - age at the time of surgery censored - indicates if an observation is censored. 1 is uncensored N)utils)dirnameabspathcCsDt}tj|ddtd}|jdddf|_|jdddf|_|S)z Load the data and return a Dataset class instance. Returns ------- Dataset instance: See DATASET_PROPOSAL.txt for more information. rN) endog_idxexog_idxdtype) _get_dataduZprocess_recarrayfloatZexogZcensors)dataZdsetr >lib/python3.7/site-packages/statsmodels/datasets/heart/data.pyload"s rcCst}tj|ddtdS)Nr)rrr)r r Zprocess_recarray_pandasr )r r r r load_pandas3s rc Cs<ttt}t|dd}tj|ddtd}WdQRX|S)Nz /heart.csvrb,T)Z delimiternamesr)rr__file__opennpZ recfromtxtr )filepathfr r r rr :s r )__doc__Z __docformat__Z COPYRIGHTZTITLEZSOURCEZ DESCRSHORTZ DESCRLONGZNOTEZnumpyrZstatsmodels.datasetsrr os.pathrrrrr r r r rs