B ZL@sBdZddlZddlmZddlmZmZm Z ddZ d ddZ dS) z[Convenience Wrappers Created on Sat Oct 30 14:56:35 2010 Author: josef-pktd License: BSD N)GLSWLSOLScCs:t|}|t|d9}||}||}||fS)zremove common rows in [y,x] that contain at least one nan TODO: this should be made more flexible, arbitrary number of arrays and 1d or 2d arrays duplicate: Skipper added sm.tools.drop_missing )npZisnanany)yxmaskr 9lib/python3.7/site-packages/statsmodels/tools/wrappers.pyremove_nanrowss r TcKsr|rt||\}}|r$tj|dd}|dk rBt||fd|i|S|dk r`t||fd|i|St||f|SdS)zget linear model with extra options for entry dispatches to regular model class and does not wrap the output If several options are exclusive, for example sigma and weights, then the chosen class depends on the implementation sequence. T)ZprependNsigmaweights)r smZ add_constantrrr)rr rrZ add_constZfilter_missingkwdsr r r linmods r)NNTT) __doc__ZnumpyrZstatsmodels.apiZapirZ statsmodelsrrrr rr r r r s