U f/e6@sdZddlZddlmZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZGdd d Zd d Zejeeed d dZejeeed ddZdS)zn Methods that can be shared by many array-like classes or subclasses: Series Index ExtensionArray N)Any)lib extract_array)!maybe_dispatch_ufunc_to_dunder_op roperator)unpack_zerodim_and_deferc@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-tiezk1ph/pandas/core/arraylike.py _cmp_methodszOpsMixin._cmp_method__eq__cCs||tjSr )roperatoreqrrrrrrszOpsMixin.__eq____ne__cCs||tjSr )rrnerrrrr"szOpsMixin.__ne____lt__cCs||tjSr )rrltrrrrr&szOpsMixin.__lt____le__cCs||tjSr )rrlerrrrr*szOpsMixin.__le____gt__cCs||tjSr )rrgtrrrrr.szOpsMixin.__gt____ge__cCs||tjSr )rrgerrrrr 2szOpsMixin.__ge__cCstSr r r rrr_logical_method9szOpsMixin._logical_method__and__cCs||tjSr )r"rand_rrrrr#<szOpsMixin.__and____rand__cCs||tjSr )r"rZrand_rrrrr%@szOpsMixin.__rand____or__cCs||tjSr )r"ror_rrrrr&DszOpsMixin.__or____ror__cCs||tjSr )r"rZror_rrrrr(HszOpsMixin.__ror____xor__cCs||tjSr )r"rxorrrrrr)LszOpsMixin.__xor____rxor__cCs||tjSr )r"rZrxorrrrrr+PszOpsMixin.__rxor__cCstSr r r rrr _arith_methodWszOpsMixin._arith_method__add__cCs||tjSr )r,raddrrrrr-ZszOpsMixin.__add____radd__cCs||tjSr )r,rZraddrrrrr/^szOpsMixin.__radd____sub__cCs||tjSr )r,rsubrrrrr0bszOpsMixin.__sub____rsub__cCs||tjSr )r,rZrsubrrrrr2fszOpsMixin.__rsub____mul__cCs||tjSr )r,rmulrrrrr3jszOpsMixin.__mul____rmul__cCs||tjSr )r,rZrmulrrrrr5nszOpsMixin.__rmul__ __truediv__cCs||tjSr )r,rtruedivrrrrr6rszOpsMixin.__truediv__ __rtruediv__cCs||tjSr )r,rZrtruedivrrrrr8vszOpsMixin.__rtruediv__ __floordiv__cCs||tjSr )r,rfloordivrrrrr9zszOpsMixin.__floordiv__Z __rfloordivcCs||tjSr )r,rZ rfloordivrrrr __rfloordiv__~szOpsMixin.__rfloordiv____mod__cCs||tjSr )r,rmodrrrrr<szOpsMixin.__mod____rmod__cCs||tjSr )r,rZrmodrrrrr>szOpsMixin.__rmod__ __divmod__cCs ||tSr )r,divmodrrrrr?szOpsMixin.__divmod__ __rdivmod__cCs||tjSr )r,rZrdivmodrrrrrAszOpsMixin.__rdivmod____pow__cCs||tjSr )r,rpowrrrrrBszOpsMixin.__pow____rpow__cCs||tjSr )r,rZrpowrrrrrDszOpsMixin.__rpow__N)#__name__ __module__ __qualname__rrrrrrrr r"r#r%r&r(r)r+r,r-r/r0r2r3r5r6r8r9r;r<r>r?rArBrDrrrrr sv                           r cCs2ddlm}t||r ||S|j|jSdS)zU Helper to check if a DataFrame is aligned with another DataFrame or Series. r DataFrameN)pandasrI isinstanceZ _indexed_samecolumnsequalsindex)framerrIrrr _is_aligneds   rP)ufuncmethodinputskwargsc sddlmddlmtfdd|D}tfdd|D}|dkr|dkrtfd d|Dtfd d|D}|rtjd td 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 rrHNDFramec3s|]}t|VqdSr rK.0xrUrr sz"_maybe_fallback..c3s|]}t|VqdSr rWrXrHrrr[sc3s|]}t|r|VqdSr rWrXrHrrr[s c3s$|]}t|rt| VqdSr )rKrPrX)rV first_framerrr[s 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)rJrIpandas.core.genericrVsumnextwarningswarn FutureWarningappendrKnpasarraygetattrr ) rQrRrSrTZ n_alignableZn_framesZ non_alignedZ new_inputsrZr)rIrVr^r_maybe_fallbacks0     rlcsddlmddlmt}tf||}|tk r>|Stf||}|tk r^|Stj j |j f}|D]P}t |do|j j k} t |dot|j |kot |j } | s| rptSqptdd|D} fdd t|| Dtd krtt| d krtd j} d d D]4} tt| | jD]\}\}}||| |<q@q,ttj| tfd dt|| D}nttjjjd krdd |D}tt|d kr|dnd }d|inifddjd krLt|d ks(jd krLtdd|D}t||}nxjd kr|tdd|D}t||}nHdkr|s|dj}|t}ntt|df|}jd krtfdd|D}n|}|S)z Compatibility with numpy ufuncs. See also -------- numpy.org/doc/stable/reference/arrays.classes.html#numpy.class.__array_ufunc__ rrU) BlockManager__array_priority____array_ufunc__css|]}t|VqdSr )typerXrrrr[szarray_ufunc..csg|]\}}t|r|qSr) issubclassrYrZtrUrr s zarray_ufunc..r]z;Cannot apply ufunc {} to mixed DataFrame and Series inputs.Nc3s,|]$\}}t|r |jfn|VqdSr )rqZreindexrr)rVreconstruct_axesrrr[0scSs g|]}t|drt|dqS)name)hasattrrkrXrrrrt8s rvcst|r|S|jjkrRdkrNjdkrJd}tj|tdd|St|St|rtj |fddi}nj |fddi}t dkr| }|S) Nouterr\zouter 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.r_r`copyFr]) rZ is_scalarndimrerfformatrgNotImplementedErrorrKZ _constructorlenZ __finalize__)resultmsg)rm alignablerRrureconstruct_kwargsrrQrr reconstruct>s0      z array_ufunc..reconstructcss|]}t|VqdSr )rirjrXrrrr[gscss|]}t|ddVqdS)T)Z extract_numpyNrrXrrrr[ks__call__c3s|]}|VqdSr rrX)rrrr[{s) rbrVZpandas.core.internalsrmrprlr rriZndarrayrorwrnrKZ_HANDLED_TYPEStuplezipr}setr|r{axes enumerateuniondictZ _AXIS_ORDERSrzZnoutrkZ_mgrapplyrj)rrQrRrSrTclsr~Zno_deferitemZhigher_priorityZhas_array_ufunctypesrobjiZax1Zax2namesrvZmgrr) rmrVrrRrrurrrQr array_ufuncsz           "&   r)__doc__rtypingrreZnumpyriZ pandas._libsrZpandas.core.constructionrZpandas.core.opsrrZpandas.core.ops.commonrr rPrQstrrlrrrrrs      >