U +ifH@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_defermaxminsumprod)maximumminimumaddmultiplyc@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)\OpsMixincCstSNNotImplementedselfotheroprH/home/mars/bis/venv/lib/python3.8/site-packages/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$.szOpsMixin.__lt____le__cCs||tjSr)rrler!rrrr&2szOpsMixin.__le____gt__cCs||tjSr)rrgtr!rrrr(6szOpsMixin.__gt____ge__cCs||tjSr)rrger!rrrr*:szOpsMixin.__ge__cCstSrrrrrr_logical_methodAszOpsMixin._logical_method__and__cCs||tjSr)r,rand_r!rrrr-DszOpsMixin.__and____rand__cCs||tjSr)r,rZrand_r!rrrr/HszOpsMixin.__rand____or__cCs||tjSr)r,ror_r!rrrr0LszOpsMixin.__or____ror__cCs||tjSr)r,rZror_r!rrrr2PszOpsMixin.__ror____xor__cCs||tjSr)r,rxorr!rrrr3TszOpsMixin.__xor____rxor__cCs||tjSr)r,rZrxorr!rrrr5XszOpsMixin.__rxor__cCstSrrrrrr _arith_method_szOpsMixin._arith_method__add__cCs||tjSr)r6rrr!rrrr7bszOpsMixin.__add____radd__cCs||tjSr)r6rZraddr!rrrr8fszOpsMixin.__radd____sub__cCs||tjSr)r6rsubr!rrrr9jszOpsMixin.__sub____rsub__cCs||tjSr)r6rZrsubr!rrrr;nszOpsMixin.__rsub____mul__cCs||tjSr)r6rmulr!rrrr<rszOpsMixin.__mul____rmul__cCs||tjSr)r6rZrmulr!rrrr>vszOpsMixin.__rmul__ __truediv__cCs||tjSr)r6rtruedivr!rrrr?zszOpsMixin.__truediv__ __rtruediv__cCs||tjSr)r6rZrtruedivr!rrrrA~szOpsMixin.__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?rArBrDrErGrHrJrKrMrrrrrsv                           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   rY)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|VqdSrr`rarQrrrdsc3s|]}t|r|VqdSrr`rarQrrrds c3s$|]}t|rt| VqdSr)rTrYra)r_ first_framerrrds 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_r nextwarningswarn FutureWarningrappendrTnpasarraygetattrr) rZr[r\r]Z n_alignableZn_framesZ non_aligned new_inputsrcr)rRr_rgr_maybe_fallbacks0     rtcs6ddlmddlmt}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)typerarrrrdszarray_ufunc..csg|]\}}t|r|qSr) issubclassrbrctr^rr s zarray_ufunc..rfz;Cannot apply ufunc {} to mixed DataFrame and Series inputs.Nc3s,|]$\}}t|r |jfn|VqdSr)ryZreindexrz)r_reconstruct_axesrrrd5scSs g|]}t|drt|dqS)name)hasattrrrrarrrr|=s r~cs(jdkr tfdd|DS|S)Nrfc3s|]}|VqdSrrra) _reconstructrrrdFsz3array_ufunc..reconstruct..)nouttuple)result)rrZrr reconstructCs z array_ufunc..reconstructcst|r|S|jjkrTdkrPjdkrLd}tj|ttd|St|St |rvj |fddi}nj |fddi}t dkr| }|S)Nouterrezouter 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.rhcopyFrf) rZ is_scalarndimrlrmformatrnrNotImplementedErrorrTZ _constructorlenZ __finalize__)rmsg)ru alignabler[r}reconstruct_kwargsrrZrrrJs8      z!array_ufunc.._reconstructoutreducecss|]}t|VqdSrrprqrarrrrdscss|]}t|ddVqdS)T)Z extract_numpyNrrarrrrds__call__)#rjr_Zpandas.core.internalsrurx_standardize_out_kwargrtrrrpZndarrayrwrrvrTZ_HANDLED_TYPESrziprsetrraxes enumerateuniondictZ _AXIS_ORDERSrdispatch_ufunc_with_outdispatch_reduction_ufuncrrrZ_mgrapplydefault_array_ufunc)rrZr[r\r]clsrZno_deferitemZhigher_priorityZhas_array_ufunctypesrobjiZax1Zax2namesr~rZmgrr) rur_rrr[r}rrrZr array_ufuncs            %   &   r)returncKs@d|kr.cs"g|]}|k r|nt|qSrrrarrrr|sz'default_array_ufunc..)anyrrr)rrZr[r\r]rsrrrrsrcOs|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. rrfrFZ numeric_onlyZaxisZskipna) AssertionErrorrrrNREDUCTION_ALIASESrrrTrrr)rrZr[r\r] method_namerrrrs      r)"__doc__rtypingrrlnumpyrpZ pandas._libsrZpandas._libs.ops_dispatchrZpandas.util._exceptionsrZpandas.core.dtypes.genericrZ pandas.corerZpandas.core.constructionr Zpandas.core.ops.commonr rrrYrZstrrtrrrrrrrrrrrs6          >/#