U
|
¬ý°dÄ> ã ! @ s` U d Z ddlmZ ddlZddlmZmZ ddlZddl m
|
Z
|
ddl m Z m Z mZ ddlmZ ddlmZmZ dd lmZmZ dd
|
lmZ dd lmZmZ dd l m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$m%Z% ddl&m'Z'm(Z(m)Z) ddl*m+Z+ ddl,m-Z-m.Z.m/Z/ ddl0m1Z1 ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z> erXddl?m@Z@mAZA ddddddddd d d d d d!d"d#hZBd$eCd%< d&d'd(d)d*d+hZDd$eCd,< d-d. ZEd[d0d1d2d3d4ZFd5d6 ZGd\d7d8d9d:d;ZHd<d=d1d>d?d@ZId<d<d<dAdBdCZJd<d0dDdEdFdGZKdHdI ZLdJdK ZMdLd;d4d%dMd,dNdOd.dIdKd6dCdPdQdRdSdTdUd dVdWd#d!d d dXd d d"dYd@dZg!ZNdS )]zD
|
Arithmetic operations for PandasObjects
|
|
This is not a public API.
|
é )Ú annotationsN)Ú TYPE_CHECKINGÚcast)Ú!maybe_dispatch_ufunc_to_dunder_op)ÚAxisÚAxisIntÚLevel)ÚAppender)Ú is_array_likeÚ is_list_like)Ú ABCDataFrameÚ ABCSeries)Úisna)Ú
|
algorithmsÚ roperator)Ú arithmetic_opÚcomp_method_OBJECT_ARRAYÚ comparison_opÚ get_array_opÚ
|
logical_opÚmaybe_prepare_scalar_for_op)Úget_op_result_nameÚunpack_zerodim_and_defer)Ú_flex_comp_doc_FRAMEÚ_op_descriptionsÚ make_flex_doc)Úinvalid_comparison)Ú
|
kleene_andÚ kleene_orÚ
|
kleene_xor)Úadd_flex_arithmetic_methods) ÚraddÚrand_ÚrdivÚrdivmodÚ rfloordivÚrmodÚrmulÚror_ÚrpowÚrsubÚrtruedivÚrxor)Ú DataFrameÚSeriesÚaddÚsubÚmulÚpowÚmodÚfloordivÚtruedivÚdivmodr! r* r' r) r& r% r+ r$ zset[str]ÚARITHMETIC_BINOPSÚeqÚneÚltÚgtÚleÚgeÚCOMPARISON_BINOPSc C s` |dk rXt | }t |}||A }| ¡ r<| ¡ } || ||@ < | ¡ rX| ¡ }||||@ < | |fS )a®
|
If a non-None fill_value is given, replace null entries in left and right
|
with this value, but only in positions where _one_ of left/right is null,
|
not both.
|
|
Parameters
|
----------
|
left : array-like
|
right : array-like
|
fill_value : object
|
|
Returns
|
-------
|
left : array-like
|
right : array-like
|
|
Notes
|
-----
|
Makes copies if fill_value is not None and NAs are present.
|
N)r ÚanyÚcopy)ÚleftÚrightÚ
|
fill_valueZ left_maskZ
|
right_maskÚmask© rE úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/ops/__init__.pyÚ
|
fill_binopp s rG Fr. Úbool)rA Úalign_asobjectc C sJ t |trB| j |j¡sB|r0| t¡} | t¡}| j|dd\} }| |fS )zalign lhs and rhs SeriesF)r@ )Ú
|
isinstancer ÚindexÚequalsZastypeÚobjectÚalign)rA rB rI rE rE rF Úalign_method_SERIES s
|
|
|
rO c s< j d¡}t|d}t|ddd fdd }||_ |S ) NÚ_Úseriesr r ©Úaxisc s¬ |d k r| |¡ t| |}t|tr8| j| ||dS t|tjttfrt |t | krbt
|
d| || j ¡}| j| ||d}||_ |S |d k r| |¡} | |S d S )N)ÚlevelrC zLengths must be equal)Ú_get_axis_numberr rJ r Z_binopÚnpÚndarrayÚlistÚtupleÚlenÚ
|
ValueErrorÚ _constructorrK ÚnameÚfillna)ÚselfÚotherrT rC rS Zres_nameÚresult©ÚoprE rF Ú flex_wrapper´ s
|
|
|
|
z(flex_method_SERIES.<locals>.flex_wrapper)NNr )Ú__name__Ústripr r )rc r] Údocrd rE rb rF Úflex_method_SERIES° s
|
rh z bool | Noner ©ÚflexrT c s: fdd}t |tjr0|jdkr0||}nþ|jdkrd}t|ddtkrTt}|jjkrxj|jj |d}n|jd jd krÀ|jd dkrÀt
|
|j¡}j|jj |d}nP|jd jd krø|jd dkrø||ddd
f }nt d j d
|
|j n |jdkr~t d |j nNt |r~t |t tfs~td d |D rvt dt|d d||}|dk rÈt |trÈ |¡s2|r¾j|d|dd\}nt dnjt |t r2 dk rè ¡nd |sj |j¡st dj|d |dd\}t| }|fS )a¯
|
Convert rhs to meet lhs dims if input is list, tuple or np.ndarray.
|
|
Parameters
|
----------
|
left : DataFrame
|
right : Any
|
axis : int, str, or None
|
flex : bool or None, default False
|
Whether this is a flex op, in which case we reindex.
|
None indicates not to check for alignment.
|
level : int or level name, default None
|
|
Returns
|
-------
|
left : DataFrame
|
right : Any
|
c s¸ d}d }t | dd tkr t} d k rt ¡dkrttjt| kr`t|jtjt| dj| j|d} n@tjt| kr¢t|jtjt| dj| j|d} | S )NzGUnable to coerce to Series, length must be {req_len}: given {given_len}ÚdtyperK )Zreq_lenZ given_len)rK rk ) ÚgetattrrM Z_get_axis_namerZ rK r[ ÚformatZ_constructor_slicedÚcolumns)rB Úmsgrk ©rS rA rE rF Ú to_seriesé s ÿÿz%align_method_FRAME.<locals>.to_seriesé é Nrk ©rK rn rk r z-Unable to coerce to DataFrame, shape must be z: given z>Unable to coerce to Series/DataFrame, dimension must be <= 2: c s s | ]}t |V qd S )N)r
|
)Ú.0ÚelrE rE rF Ú <genexpr>- s z%align_method_FRAME.<locals>.<genexpr>zUnable to coerce list of z to Series/DataFrameÚouterF)ÚjoinrT r@ zOCan only compare identically-labeled (both index and columns) DataFrame objectszdOperands are not aligned. Do `left, right = left.align(right, axis=1, copy=False)` before operating.)ry rS rT r@ )rJ rV rW Úndimrl rM Úshaper\ rK rn Ú broadcast_tor[ r r r r? ÚtypeZ _indexed_samerN rU ZaxesrL Ú _maybe_align_series_as_frame)rA rB rS rj rT rq rk rE rp rF Úalign_method_FRAMEÓ sz
|
|
ÿ" ÿ"ÿ
|
ÿÿ ÿ ÿ ÿ
|
r r- Úint)rA rS Úreturnc C s t | tst|tjks"|tjkr&dS t |ts4dS |dkr|dkr|dkr| j ¡ }|j ¡ }| |¡}t
|
|rt
|
|t
|
|krt
|
|t
|
|ksdS dS )zU
|
Check if this is an operation between DataFrames that will need to reindex.
|
FNrr T) rJ r ÚAssertionErrorÚoperatorr2 r r) rn ÚuniqueÚ intersectionrZ ) rA rB rc rS rC rT Z left_uniquesZ right_uniquesÚcolsrE rE rF Úshould_reindex_frame_opQ s
|
|
|
|
ÿÿr )rA rB r c C s² | j j|j dddd\}}}| jdd
|f }|jdd
|f }|||}| j j|j dddd\} }
|
}
|
|j jr |j | ¡\} }
|
t | ¡} |jd| | gidd}n|j| dd}|S ) a
|
For DataFrame-with-DataFrame operations that require reindexing,
|
operate only on shared columns, then reindex.
|
|
Parameters
|
----------
|
left : DataFrame
|
right : DataFrame
|
op : binary operator
|
|
Returns
|
-------
|
DataFrame
|
ÚinnerNT)ÚhowrT Zreturn_indexersrx rr )Z
|
allow_dupsrR ) rn ry ZilocZhas_duplicatesZget_indexer_non_uniquer Zunique1dZ_reindex_with_indexersZreindex) rA rB rc r ZlcolsZrcolsZnew_leftZ new_rightra Z join_columnsrP ZindexerrE rE rF Ú frame_arith_method_with_reindexp s0 ÿ
|
ÿ
|
|
ÿr r )ÚframerQ rS c C sv |j }t|tjs,|jdkr(t |¡}n|S |dkrB| dd¡}n | dd¡}t || j¡}t | || j
|
| j |jdS )zb
|
If the Series operand is not EA-dtype, we can broadcast to 2D and operate
|
blockwise.
|
)zdatetime64[ns]ztimedelta64[ns]r éÿÿÿÿrr rt ) Z_valuesrJ rV rW rk ZasarrayZreshaper| r{ r} rK rn )r rQ rS ZrvaluesrE rE rF r~ s
|
ÿr~ c sF j d¡}t t|d}t|ddd fdd }||_ |S ) NrP Z dataframern r rR c sà |d k r| |¡nd}tt|}t| ||||r>t| |S t|tr`|d k r`td| dt|| j }t
|
| ||d|d\} }t|t r| | |¡}n:t|tr¸| j ||d}n |d k rÊ| |¡} | |¡}| |¡S )Nrr z fill_value z not supported.Tri rR )rU r r r r rJ r ÚNotImplementedErrorr r{ r r Z_combine_frameÚ_dispatch_frame_opr^ Ú_construct_result)r_ r` rS rT rC Únew_data©Zna_oprc rE rF Úf¼ s
|
|
|
|
z"flex_arith_method_FRAME.<locals>.f)rn NN)re rf r r r ©rc Úop_namerg r rE r rF Úflex_arith_method_FRAME¶ s
|
r c sH j d¡}tj|t| d d}t|d dd fdd }||_ |S )
|
NrP Údesc)r r rn r rR c sF |d k r| |¡nd}t| ||d|d\} }| j| |d}| |¡S )Nrr Tri rR )rU r r r )r_ r` rS rT r rb rE rF r ç s z!flex_comp_method_FRAME.<locals>.f)rn N)re rf r rm r r r rE rb rF Úflex_comp_method_FRAMEà s
|
ÿr r r r r r r r r r r r" r# r( r, r )F)FN)OÚ__doc__Ú
|
__future__r r Útypingr r ÚnumpyrV Zpandas._libs.ops_dispatchr Zpandas._typingr r r Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr
|
r Zpandas.core.dtypes.genericr r Zpandas.core.dtypes.missingr Z pandas.corer r Zpandas.core.ops.array_opsr r r r r r Zpandas.core.ops.commonr r Zpandas.core.ops.docstringsr r r Zpandas.core.ops.invalidr Zpandas.core.ops.mask_opsr r r Zpandas.core.ops.methodsr Zpandas.core.roperatorr! r" r# r$ r% r& r' r( r) r* r+ r, Zpandasr- r. r7 Ú__annotations__r> rG rO rh r r r r~ r r Ú__all__rE rE rE rF Ú<module> s¦ 8ð -$ ÿ~ ,*ß
|
