a GbH@s"dZddlZddlmZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZd d d ddZGdddZddZejeeedddZejeeedddZedddZejedddZdddd Zejedd!d"Z ejedd#d$Z!dS)%zn Methods that can be shared by many array-like classes or subclasses: Series Index ExtensionArray N)Any)lib)!maybe_dispatch_ufunc_to_dunder_op)find_stack_level) ABCNDFrame) roperator extract_array)unpack_zerodim_and_defermaxminsumZprod)ZmaximumZminimumaddZmultiplyc@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)\OpsMixincCstSNNotImplementedselfotheropr4lib/python3.9/site-packages/pandas/core/arraylike.py _cmp_method#szOpsMixin._cmp_method__eq__cCs||tjSr)roperatoreqrrrrrr&szOpsMixin.__eq____ne__cCs||tjSr)rrnerrrrr*szOpsMixin.__ne____lt__cCs||tjSr)rrltrrrrr .szOpsMixin.__lt____le__cCs||tjSr)rrlerrrrr"2szOpsMixin.__le____gt__cCs||tjSr)rrgtrrrrr$6szOpsMixin.__gt____ge__cCs||tjSr)rrgerrrrr&:szOpsMixin.__ge__cCstSrrrrrr_logical_methodAszOpsMixin._logical_method__and__cCs||tjSr)r(rand_rrrrr)DszOpsMixin.__and____rand__cCs||tjSr)r(rZrand_rrrrr+HszOpsMixin.__rand____or__cCs||tjSr)r(ror_rrrrr,LszOpsMixin.__or____ror__cCs||tjSr)r(rZror_rrrrr.PszOpsMixin.__ror____xor__cCs||tjSr)r(rxorrrrrr/TszOpsMixin.__xor____rxor__cCs||tjSr)r(rZrxorrrrrr1XszOpsMixin.__rxor__cCstSrrrrrr _arith_method_szOpsMixin._arith_method__add__cCs||tjSr)r2rrrrrrr3bszOpsMixin.__add____radd__cCs||tjSr)r2rZraddrrrrr4fszOpsMixin.__radd____sub__cCs||tjSr)r2rsubrrrrr5jszOpsMixin.__sub____rsub__cCs||tjSr)r2rZrsubrrrrr7nszOpsMixin.__rsub____mul__cCs||tjSr)r2rmulrrrrr8rszOpsMixin.__mul____rmul__cCs||tjSr)r2rZrmulrrrrr:vszOpsMixin.__rmul__ __truediv__cCs||tjSr)r2rtruedivrrrrr;zszOpsMixin.__truediv__ __rtruediv__cCs||tjSr)r2rZrtruedivrrrrr=~szOpsMixin.__rtruediv__ __floordiv__cCs||tjSr)r2rfloordivrrrrr>szOpsMixin.__floordiv__Z __rfloordivcCs||tjSr)r2rZ rfloordivrrrr __rfloordiv__szOpsMixin.__rfloordiv____mod__cCs||tjSr)r2rmodrrrrrAszOpsMixin.__mod____rmod__cCs||tjSr)r2rZrmodrrrrrCszOpsMixin.__rmod__ __divmod__cCs ||tSr)r2divmodrrrrrDszOpsMixin.__divmod__ __rdivmod__cCs||tjSr)r2rZrdivmodrrrrrFszOpsMixin.__rdivmod____pow__cCs||tjSr)r2rpowrrrrrGszOpsMixin.__pow____rpow__cCs||tjSr)r2rZrpowrrrrrIszOpsMixin.__rpow__N)#__name__ __module__ __qualname__rr rrr r"r$r&r(r)r+r,r.r/r1r2r3r4r5r7r8r:r;r=r>r@rArCrDrFrGrIrrrrrsv                           rcCs2ddlm}t||r ||S|j|jSdS)zU Helper to check if a DataFrame is aligned with another DataFrame or Series. r DataFrameN)pandasrN isinstanceZ _indexed_samecolumnsZequalsindex)framerrNrrr _is_aligneds   rT)ufuncmethodinputskwargsc sddlmddlmtfdd|D}tfdd|D}|dkr|dkrtfd d|Dtfd d|D}|rtjd tt d g}|D]>}|ur| |qt |r| t |q| |qt|||i|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 rrMNDFramec3s|]}t|VqdSrrP.0xrYrr z"_maybe_fallback..c3s|]}t|VqdSrr[r\rMrrr_r`c3s|]}t|r|VqdSrr[r\rMrrr_r`c3s$|]}t|rt| VqdSr)rPrTr\)rZ first_framerrr_saCalling 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)rOrNpandas.core.genericrZr nextwarningswarn FutureWarningrappendrPnpasarraygetattrr) rUrVrWrXZ n_alignableZn_framesZ non_aligned new_inputsr^r)rNrZrcr_maybe_fallbacks0     rpcs`ddlmddlmt}tfi|}tg|Ri|}|turR|Stg|Ri|}|turx|St j j |j f}|D]P}t |do|j j k} t |dot|j |vot|j } | s| rtSqtdd|D} fdd t|| Dtd krtt| d kr4td j} d d D]4} tt| | jD]\}\}}||| |<qZqFttj| tfd dt|| D}nttjjjd krdd |D}tt|d kr|dnd }d|inifdd}fddd|vrVtg|Ri|}||Sdkrtg|Ri|}|tur|Sjd krt|d ksjd krtdd|D}t|i|}n~jd kr tdd|D}t|i|}nJdkr6|s6|dj }|!t}nt"|dg|Ri|}||}|S)z Compatibility with numpy ufuncs. See also -------- numpy.org/doc/stable/reference/arrays.classes.html#numpy.class.__array_ufunc__ rrY) BlockManager__array_priority____array_ufunc__css|]}t|VqdSr)typer\rrrr_r`zarray_ufunc..csg|]\}}t|r|qSr) issubclassr]r^trYrr r`zarray_ufunc..rbz;Cannot apply ufunc {} to mixed DataFrame and Series inputs.Nc3s0|](\}}t|r$|jfin|VqdSr)ruZreindexrv)rZreconstruct_axesrrr_5scSs g|]}t|drt|dqS)name)hasattrrnr\rrrrx=r`rzcs(jdkr tfdd|DS|S)Nrbc3s|]}|VqdSrrr\) _reconstructrrr_Fr`z3array_ufunc..reconstruct..)nouttuple)result)r|rUrr reconstructCs z array_ufunc..reconstructcst|r|S|jjkrTdkrPjdkrLd}tj|ttd|St|St |rzj |fiddi}nj |fiddi}t dkr| }|S)NZouterrazouter 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.rdcopyFrb) rZ is_scalarndimrhriformatrjrNotImplementedErrorrPZ _constructorlenZ __finalize__)rmsg)rq alignablerVryreconstruct_kwargsrrUrrr|Js6      z!array_ufunc.._reconstructoutreducecss|]}t|VqdSrrlrmr\rrrr_r`css|]}t|ddVqdS)T)Z extract_numpyNrr\rrrr_r`__call__)#rfrZZpandas.core.internalsrqrt_standardize_out_kwargrprrrlZndarrayrsr{rrrPZ_HANDLED_TYPESr~ziprsetrraxes enumerateuniondictZ _AXIS_ORDERSrdispatch_ufunc_with_outdispatch_reduction_ufuncr}rnZ_mgrZapplydefault_array_ufunc)rrUrVrWrXclsrZno_deferitemZhigher_priorityZhas_array_ufunctypesrobjiZax1Zax2namesrzrZmgrr) rqrZr|rrVryrrrUr array_ufuncs           %   &   r)returncKs@d|vr.cs"g|]}|ur|nt|qSrrr\rrrrxr`z'default_array_ufunc..)anyrrn)rrUrVrWrXrorrrrsrcOs|dks Jt|dks$|d|ur(tS|jtvr6tSt|j}t||sNtS|jdkrzt|trjd|d<d|vrzd|d<t||fddi|S)z@ Dispatch ufunc reductions to self's reduction methods. rrbrFZ numeric_onlyZaxisZskipna) rrrJREDUCTION_ALIASESr{rrPrrn)rrUrVrWrX method_namerrrrs      r)"__doc__rtypingrrhZnumpyrlZ pandas._libsrZpandas._libs.ops_dispatchrZpandas.util._exceptionsrZpandas.core.dtypes.genericrZ pandas.corerZpandas.core.constructionr Zpandas.core.ops.commonr rrrTrUstrrprrrrrrrrrrrs6          >/#