B ZX @s`dZdZdZdZdZdZdZdZdd lZ dd l m Z dd l mZmZd d ZddZddZd S)z Fair's Extramarital Affairs DataZrestructuredtextz'Included with permission of the author.zAffairs datasetz Fair, Ray. 1978. "A Theory of Extramarital Affairs," `Journal of Political Economy`, February, 45-61. The data is available at http://fairmodel.econ.yale.edu/rayfair/pdf/2011b.htm zExtramarital affair data.zExtramarital affair data used to explain the allocation of an individual's time among work, time spent with a spouse, and time spent with a paramour. The data is used as an example of regression with censored data.aL:: Number of observations: 6366 Number of variables: 9 Variable name definitions: rate_marriage : How rate marriage, 1 = very poor, 2 = poor, 3 = fair, 4 = good, 5 = very good age : Age yrs_married : No. years married. Interval approximations. See original paper for detailed explanation. children : No. children religious : How relgious, 1 = not, 2 = mildly, 3 = fairly, 4 = strongly educ : Level of education, 9 = grade school, 12 = high school, 14 = some college, 16 = college graduate, 17 = some graduate school, 20 = advanced degree occupation : 1 = student, 2 = farming, agriculture; semi-skilled, or unskilled worker; 3 = white-colloar; 4 = teacher counselor social worker, nurse; artist, writers; technician, skilled worker, 5 = managerial, administrative, business, 6 = professional with advanced degree occupation_husb : Husband's occupation. Same as occupation. affairs : measure of time spent in extramarital affairs See the original paper for more details. N)utils)dirnameabspathcCst}tj|ddtdS)z Load the data and return a Dataset class instance. Returns ------- Dataset instance: See DATASET_PROPOSAL.txt for more information. N) endog_idxexog_idxdtype) _get_dataduZprocess_recarrayfloat)datar =lib/python3.7/site-packages/statsmodels/datasets/fair/data.pyload9s rcCst}tj|ddtdS)Nr)rrr)r r Zprocess_recarray_pandasr )r r r r load_pandasGs rc Cs<ttt}t|dd}tj|ddtd}WdQRX|S)Nz /fair.csvrb,T)Z delimiternamesr)rr__file__opennpZ recfromtxtr )filepathfr r r rr Ns r )__doc__Z __docformat__Z COPYRIGHTZTITLEZSOURCEZ DESCRSHORTZ DESCRLONGZNOTEZnumpyrZstatsmodels.datasetsrr os.pathrrrrr r r r rs