3 pd_4@sUdZddlmZddlmZddlmZmZmZm Z ddl m Z m Z ddlmZmZddlmZddlmZmZmZGd d d Zed d Zeed dddZeeddddZddZddZddZeZddaed<ded<d ed<ee edkrd ed<eeddddZ eZ!d !dbe!d<d e!d<ee!ddddZ"d!d"Z#ed d Z$ee%ef$ee$d#dd$dZ&d%d&Z'eZ(d'(d e(d(<d e(d)<ee(ddd*Z)ee(d+dddZ*d,d-Z+eZ,d.,d e,d(<d e,d)<d/e,d0<ee,d1dddZ-ee,d2dddZ.ed d/d3Z/ee/d4d5Z0ed d d/d6Z1ee1d7dddZ2ee1d8dddZ3ed9d:Z4ee%e%f4ee4d;dddZ5ed d<Z6ee%ef6ee6d=dddZ7ed d Z8ee%ef8ee8d>dddZ9eZ:d:dce:d<de:d<d e:d<ee:d?d4d@Z;eZd e>d<d/e>d0<d e>dB<edDdddZBee?dEdddZCee/tmp/pip-build-7vycvbft/pandas/pandas/compat/numpy/function.py__init__$szCompatValidator.__init__)rreturncCs|s|r|dkr|jn|}|dkr(|jn|}|dkr:|jn|}|dkrXt||||jnD|dkrpt|||jn,|dkrt|||||jntd|ddS)Nargskwargsbothzinvalid validation method '')rrrr rrr ValueError)rrrrrrrrr__call__0szCompatValidator.__call__)NNN)NNN)__name__ __module__ __qualname__rstrrrrrrrr#s rN)outZargminr)rrrZargmaxcCs(t|ts|dkr |f|}d}||fS)NT) isinstancer )skipnarrrrprocess_skipnaVs r(cCst||\}}t|||S)a! If 'Series.argmin' is called via the 'numpy' library, the third parameter in its signature is 'out', which takes either an ndarray or 'None', so check if the 'skipna' parameter is either an instance of ndarray or is None, since 'skipna' itself should be a boolean )r(validate_argmin)r'rrrrrvalidate_argmin_with_skipna^s r*cCst||\}}t|||S)a! If 'Series.argmax' is called via the 'numpy' library, the third parameter in its signature is 'out', which takes either an ndarray or 'None', so check if the 'skipna' parameter is either an instance of ndarray or is None, since 'skipna' itself should be a boolean )r(validate_argmax)r'rrrrrvalidate_argmax_with_skipnaks r,z+OrderedDict[str, Optional[Union[int, str]]]axisZ quicksortkindorderz1.17.0Zargsort)rrrzOrderedDict[str, Optional[int]]cCs0t|s|dkr|f|}d}t||dd|S)a% If 'Categorical.argsort' is called via the 'numpy' library, the first parameter in its signature is 'axis', which takes either an integer or 'None', so check if the 'ascending' parameter has either integer type or is None, since 'ascending' itself should be a boolean NT)r)r validate_argsort_kind)Z ascendingrrrrrvalidate_argsort_with_ascendings  r2Zclipr0cCs&t|tr|f|}d}t|||S)a If 'NDFrame.clip' is called via the numpy library, the third parameter in its signature is 'out', which can takes an ndarray, so check if the 'axis' parameter is an instance of ndarray, since 'axis' itself should either be an integer or None N)r&r validate_clip)r-rrrrrvalidate_clip_with_axiss    r4zOrderedDict[str, Any]dtyper$)rrZcumsumcCs(t|s|f|}d}t|||d|S)z If this function is called via the 'numpy' library, the third parameter in its signature is 'dtype', which takes either a 'numpy' dtype or 'None', so check if the 'skipna' parameter is a boolean or not T)r)r validate_cum_func)r'rrnamerrrvalidate_cum_func_with_skipnas  r8z OrderedDict[str, Optional[bool]]Fkeepdimsallany)r$r9r)r)r-r$r9minmaxC)r/Zreshape)r-repeatroundsort)rrzOrderedDict[str, Optional[Any]]initialZoverwrite_inputsumprodmeanZmedianzOrderedDict[str, Optional[str]]raisemodeZtakecCs4t|ts|dkr |f|}d}t||ddd|S)z If this function is called via the 'numpy' library, the third parameter in its signature is 'axis', which takes either an ndarray or 'None', so check if the 'convert' parameter is either an instance of ndarray or is None NTr0r)rr)r&r validate_take)convertrrrrrvalidate_take_with_convert(s  rJ)ZaxesZ transpose)rcCsFd}d|d}t|dkr$t|x|D]}||kr*t|q*WdS)Nr-r5r$z9numpy operations are not valid with window objects. Use .z() directly instead r)r-r5r$)lenr )r7rr numpy_argsmsgargrrrvalidate_window_func=s   rOcCsFd}d|d}t|dkr$t|x|D]}||kr*t|q*WdS)Nr-r5r$zFnumpy operations are not valid with window objects. Use .rolling(...).z () instead r)r-r5r$)rKr )r7rrrLrMrNrrrvalidate_rolling_funcLs   rPcCsFd}d|d}t|dkr$t|x|D]}||kr*t|q*WdS)Nr-r5r$zHnumpy operations are not valid with window objects. Use .expanding(...).z () instead r)r-r5r$)rKr )r7rrrLrMrNrrrvalidate_expanding_func[s   rQcCsD|dkr g}t|t|}t|t|dkr@td|ddS)z 'args' and 'kwargs' should be empty, except for allowed kwargs because all of their necessary parameters are explicitly listed in the function signature Nrz?numpy operations are not valid with groupby. Use .groupby(...).z () instead)setrKr )r7rrallowedrrrvalidate_groupby_funcjs rTstdvar)rrcCs:t|t|dkr6|tkr.td|dntddS)z 'args' and 'kwargs' should be empty because all of their necessary parameters are explicitly listed in the function signature rzAnumpy operations are not valid with resample. Use .resample(...).z () insteadztoo many arguments passed inN)rKRESAMPLER_NUMPY_OPSr TypeError)rrrrrrvalidate_resampler_funcs rY)r-rcCs@d}|dkrdS||ks,|dkr<||dkrs   *