a \:b@slddlmmZddlmZmZddlZiZ GdddeZdddZ dd Z d d Z d d Z ddZdS)N) dmatricesNAActionc@seZdZddZdS)rcsHtj|djdtd}|D] }||O}q|||_fdd|DS)Nr)Zdtypecsg|]}|dfqS).).0vZ good_maskr?lib/python3.9/site-packages/statsmodels/formula/formulatools.py z,NAAction._handle_NA_drop..)npZzerosshapebool missing_mask)selfvaluesZis_NAsZoriginsZ total_maskZis_NArrr_handle_NA_drops  zNAAction._handle_NA_dropN)__name__ __module__ __qualname__rrrrrr srdropc Cst|ttrtt|St|d}|durlt||rTt|||f|d|d}qt|||f|d|d}n2t|drt|||d|d}nt|||d|d}t |dd}t |sd}t |dkr|dj }nd}|||fS)a; Returns endog, exog, and the model specification from arrays and formula. Parameters ---------- Y : array_like Either endog (the LHS) of a model specification or all of the data. Y must define __getitem__ for now. X : array_like Either exog or None. If all the data for the formula is provided in Y then you must explicitly set X to None. formula : str or patsy.model_desc You can pass a handler by import formula_handler and adding a key-value pair where the key is the formula object class and the value is a function that returns endog, exog, formula object. Returns ------- endog : array_like Should preserve the input type of Y,X. exog : array_like Should preserve the input type of Y,X. Could be None. )Zon_NANZ dataframe)Z return_typeZ NA_actionr) isinstancetupleformula_handlerkeystyper data_utilZ_is_using_pandasrgetattrr anylen design_info) YXZformulaZdepthZmissingZ na_actionresultrr rrrhandle_formula_datas2            r$cCs"ddlm}||vr|||S)z, Remove intercept from Patsy terms. r INTERCEPT) patsy.descr&remove)termsr&rrr_remove_intercept_patsyQs  r*cCsddlm}||jvS)Nrr%)r'r&r))r r&rrr_has_intercept[s r+cs0ddlmddlm}|fdd|jDS)zR Returns boolean array index indicating which column holds the intercept. rr%)arraycsg|] }|kqSrr)rir%rrr fr z"_intercept_idx..)r'r&numpyr,r))r r,rr%r_intercept_idx`s  r/cCs"ddlm}|jj}|||}|S)z r)linear_constraint)Zpatsy.constraintr0Zmodel exog_names)Z model_resultsZ test_formular0r1ZLCrrrmake_hypotheses_matricesis  r2)rr)Zstatsmodels.tools.dataZtoolsdatarZpatsyrrr.r rr$r*r+r/r2rrrrs 9