U sVcI@s$dZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZd d dddZGdddZddZdddddddZdddddddZdddd Zddd!d"d#Zd$dd%d&Zddd!d'd(Zddd!d)d*Z dS)+zn Methods that can be shared by many array-like classes or subclasses: Series Index ExtensionArray ) annotationsN)Any)lib)!maybe_dispatch_ufunc_to_dunder_op)find_stack_level) ABCNDFrame) roperator extract_array)unpack_zerodim_and_defermaxminsumprod)maximumZminimumaddmultiplyc@seZdZddZedddZedddZed d d Zed d dZedddZ edddZ ddZ edddZ edddZ edddZed d!d"Zed#d$d%Zed&d'd(Zd)d*Zed+d,d-Zed.d/d0Zed1d2d3Zed4d5d6Zed7d8d9Zed:d;d<Zed=d>d?Zed@dAdBZedCdDdEZedFdGdHZedIdJdKZedLdMdNZedOdPdQZedRdSdTZ edUdVdWZ!edXdYdZZ"d[S)\OpsMixincCstSNNotImplementedselfotheropr9/tmp/pip-unpacked-wheel-xj8nt62q/pandas/core/arraylike.py _cmp_method%szOpsMixin._cmp_method__eq__cCs||tjSr)roperatoreqrrrrrr(szOpsMixin.__eq____ne__cCs||tjSr)rrner!rrrr",szOpsMixin.__ne____lt__cCs||tjSr)rrltr!rrrr$0szOpsMixin.__lt____le__cCs||tjSr)rrler!rrrr&4szOpsMixin.__le____gt__cCs||tjSr)rrgtr!rrrr(8szOpsMixin.__gt____ge__cCs||tjSr)rrger!rrrr*<szOpsMixin.__ge__cCstSrrrrrr_logical_methodCszOpsMixin._logical_method__and__cCs||tjSr)r,rand_r!rrrr-FszOpsMixin.__and____rand__cCs||tjSr)r,rZrand_r!rrrr/JszOpsMixin.__rand____or__cCs||tjSr)r,ror_r!rrrr0NszOpsMixin.__or____ror__cCs||tjSr)r,rZror_r!rrrr2RszOpsMixin.__ror____xor__cCs||tjSr)r,rxorr!rrrr3VszOpsMixin.__xor____rxor__cCs||tjSr)r,rZrxorr!rrrr5ZszOpsMixin.__rxor__cCstSrrrrrr _arith_methodaszOpsMixin._arith_method__add__cCs||tjSr)r6rrr!rrrr7dszOpsMixin.__add____radd__cCs||tjSr)r6rZraddr!rrrr8hszOpsMixin.__radd____sub__cCs||tjSr)r6rsubr!rrrr9lszOpsMixin.__sub____rsub__cCs||tjSr)r6rZrsubr!rrrr;pszOpsMixin.__rsub____mul__cCs||tjSr)r6rmulr!rrrr<tszOpsMixin.__mul____rmul__cCs||tjSr)r6rZrmulr!rrrr>xszOpsMixin.__rmul__ __truediv__cCs||tjSr)r6rtruedivr!rrrr?|szOpsMixin.__truediv__ __rtruediv__cCs||tjSr)r6rZrtruedivr!rrrrAszOpsMixin.__rtruediv__ __floordiv__cCs||tjSr)r6rfloordivr!rrrrBszOpsMixin.__floordiv__Z __rfloordivcCs||tjSr)r6rZ rfloordivr!rrr __rfloordiv__szOpsMixin.__rfloordiv____mod__cCs||tjSr)r6rmodr!rrrrEszOpsMixin.__mod____rmod__cCs||tjSr)r6rZrmodr!rrrrGszOpsMixin.__rmod__ __divmod__cCs ||tSr)r6divmodr!rrrrHszOpsMixin.__divmod__ __rdivmod__cCs||tjSr)r6rZrdivmodr!rrrrJszOpsMixin.__rdivmod____pow__cCs||tjSr)r6rpowr!rrrrKszOpsMixin.__pow____rpow__cCs||tjSr)r6rZrpowr!rrrrMszOpsMixin.__rpow__N)#__name__ __module__ __qualname__rr rr"r$r&r(r*r,r-r/r0r2r3r5r6r7r8r9r;r<r>r?rArBrDrErGrHrJrKrMrrrrr!sv                           rcCs2ddlm}t||r ||S|j|jSdS)zU Helper to check if a DataFrame is aligned with another DataFrame or Series. r DataFrameN)pandasrR isinstanceZ _indexed_samecolumnsequalsindex)framerrRrrr _is_aligneds   rYznp.ufuncstrr)ufuncmethodinputskwargsc sddlmddlmtfdd|D}tfdd|D}|dkr|dkrtfd d|Dtfd d|D}|rtjd tt d g}|D]>}|kr| |qt |r| t |q| |qt||||StS) a In the future DataFrame, inputs to ufuncs will be aligned before applying the ufunc, but for now we ignore the index but raise a warning if behaviour would change in the future. This helper detects the case where a warning is needed and then fallbacks to applying the ufunc on arrays to avoid alignment. See https://github.com/pandas-dev/pandas/pull/39239 rrQNDFramec3s|]}t|VqdSrrT.0xr_rr sz"_maybe_fallback..c3s|]}t|VqdSrrarbrQrrresc3s|]}t|r|VqdSrrarbrQrrres c3s$|]}t|rt| VqdSr)rTrYrb)r` first_framerrres aCalling a ufunc on non-aligned DataFrames (or DataFrame/Series combination). Currently, the indices are ignored and the result takes the index/columns of the first DataFrame. In the future , the DataFrames/Series will be aligned before applying the ufunc. Convert one of the arguments to a NumPy array (eg 'ufunc(df1, np.asarray(df2)') to keep the current behaviour, or align manually (eg 'df1, df2 = df1.align(df2)') before passing to the ufunc to obtain the future behaviour and silence this warning. stacklevel)rSrRpandas.core.genericr`rnextwarningswarn FutureWarningrappendrTnpasarraygetattrr) r[r\r]r^Z n_alignableZn_framesZ non_aligned new_inputsrdr)rRr`rhr_maybe_fallbacks0     rucs6ddlmddlmt}tf|}tf||}|tk rH|Stf||}|tk rh|St j j |j f}|D]P}t |do|j j k} t |dot|j |kot|j } | s| rztSqztdd|D} fdd t|| Dtd krtt| d kr$td j} d d D]4} tt| | jD]\}\}}||| |<qJq6ttj| tfd dt|| D}nttjjjd krdd |D}tt|d kr|dnd }d|inifdd}fddd|kr@tf||}||Sdkrltf||}|tk rl|Sjd krt|d ksjd krtdd|D}t||}ntjd krtdd|D}t||}nDdkr|s|dj }|!t}nt"|df||}||}|S)z Compatibility with numpy ufuncs. See also -------- numpy.org/doc/stable/reference/arrays.classes.html#numpy.class.__array_ufunc__ rr_) BlockManager__array_priority____array_ufunc__css|]}t|VqdSr)typerbrrrre!szarray_ufunc..csg|]\}}t|r|qSr) issubclassrcrdtr_rr "s zarray_ufunc..rgz;Cannot apply ufunc {} to mixed DataFrame and Series inputs.Nc3s,|]$\}}t|r |jfn|VqdSr)rzZreindexr{)r`reconstruct_axesrrre:scSs g|]}t|drt|dqS)name)hasattrrsrbrrrr}Bs rcs(jdkr tfdd|DS|S)Nrgc3s|]}|VqdSrrrb) _reconstructrrreKsz3array_ufunc..reconstruct..)nouttuple)result)rr[rr reconstructHs z array_ufunc..reconstructcst|r|S|jjkrTdkrPjdkrLd}tj|ttd|St|St |rvj |fddi}nj |fddi}t dkr| }|S)Nouterrfzouter method for ufunc {} is not implemented on pandas objects. Returning an ndarray, but in the future this will raise a 'NotImplementedError'. Consider explicitly converting the DataFrame to an array with '.to_numpy()' first.ricopyFrg) rZ is_scalarndimrmrnformatrorNotImplementedErrorrTZ _constructorlenZ __finalize__)rmsg)rv alignabler\r~reconstruct_kwargsrr[rrrOs8      z!array_ufunc.._reconstructoutreducecss|]}t|VqdSrrqrrrbrrrrescss|]}t|ddVqdS)T)Z extract_numpyNr rbrrrres__call__)#rkr`Zpandas.core.internalsrvry_standardize_out_kwargrurrrqZndarrayrxrrwrTZ_HANDLED_TYPESrziprsetrraxes enumerateuniondictZ _AXIS_ORDERSrdispatch_ufunc_with_outdispatch_reduction_ufuncrrsZ_mgrapplydefault_array_ufunc)rr[r\r]r^clsrZno_deferitemZhigher_priorityZhas_array_ufunctypesrobjiZax1Zax2namesrrZmgrr) rvr`rrr\r~rrr[r array_ufuncs            %   &   rr)returncKs@d|kr.cs"g|]}|k r|nt|qSrrrbrrrr}sz'default_array_ufunc..)anyrrs)rr[r\r]r^rtrrrrsrcOs|dks tt|dks$|d|k r(tS|jtkr6tSt|j}t||sNtS|jdkrzt|trjd|d<d|krzd|d<t ||fddi|S)z@ Dispatch ufunc reductions to self's reduction methods. rrgrFZ numeric_onlyZaxisZskipna) AssertionErrorrrrNREDUCTION_ALIASESrrrTrrs)rr[r\r]r^ method_namerrrrs      r)!__doc__ __future__rrtypingrrmZnumpyrqZ pandas._libsrZpandas._libs.ops_dispatchrZpandas.util._exceptionsrZpandas.core.dtypes.genericrZ pandas.corerZpandas.core.constructionr Zpandas.core.ops.commonr rrrYrurrrrrrrrrrs8           >2#