*jL$dZddlmZddlZddlmZddlZddlmZm Z ddl Z ddl m Z mZmZmZmZddlmZmZmZmZddlmZmZdd lmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(dd l)m*Z*dd l+m,Z,ddl-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5erddl6m7Z7m8Z8dZ9dZ:ddZ;dd dZd#dZ?d"dZ@dZAd$dZBeje*jeje*jeje*jeIe*jhZKd%dZLy)&zY Functions for arithmetic and comparison operations on NumPy arrays and ExtensionArrays. ) annotationsN)partial) TYPE_CHECKINGAny)NaT Timedelta Timestamplibops) BaseOffsetget_supported_dtypeis_supported_dtype is_unitless)'construct_1d_object_array_from_listlikefind_common_type) ensure_object is_bool_dtype is_list_likeis_numeric_v_string_likeis_object_dtype is_scalar)ABCExtensionArrayABCIndex ABCSeries)isnanotna) roperator) expressions)ensure_wrapped_if_datetimelikesanitize_array)missing)should_extension_dispatch)invalid_comparison) ArrayLikeShapec|kt|}t|}||z }|jr|j}||||z<|jr|j}||||z<||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. )ranycopy)leftright fill_value left_mask right_maskmasks ;C:\FundamentalWorkbench\vendor\pandas\core\ops\array_ops.py fill_binopr0Jsy*J %[ :% ==?99;D%/DT! " >> JJLE'1E*t# $ ;ct|tr t|}t|tjt t frt|js|jtj}t|t t fr |j}|j|jk7r!td|j|jtj|j!|j!|}n%tj"|j!||}|j%|jS)NzShapes must match) isinstancelistrnpndarrayrrrdtypeastypeobject__valuesshape ValueErrorlibops vec_compareravelscalar_comparereshape)opxyresults r/comp_method_OBJECT_ARRAYrFss!T 3A 6!bjj)X67qww'$A a)X. / A 77agg 0!''177C C##AGGIqwwy"=&&qwwy!R8 >>!'' ""r1c|j}t|tjrt |j |j g}tj |j|}t|t|k7r t|j|jt|}|j}t||jz}|jr|||||||<nt|stdt|tj |j|j }t|}|t urtj"|dk(d|}n,|t$j&urtj"|dk(d|}|jr||||||<tj(||tj*|j-|j}|S)z If the given arithmetic operation fails, attempt it again on only the non-null elements of the input array(s). Parameters ---------- x : np.ndarray y : np.ndarray, Series, Index op : binary operator r7z1Cannot broadcast np.ndarray with operand of type F)r?r3r5r6rr7emptysizelenr<r;rr'r TypeErrortypepowwhererrpowputmasknanrA) rCrDrBxravr7rEymaskyravr.s r/_masked_arith_oprWs 779D!RZZ  !''177!34!&&. q6SV QWWagg. .awwyT{U[[]* 88:d4j$t*5F4L|CDG9M  !&&0T{ 988AFE40D 9>> !88AFE40D 88:d4j!,F4LJJvubff% ^^AGG $F Mr1crt|tr|}nttj|} |||}|r t|s|tur t|||Stj||||S#t $r;|s5|j tk(st|ddtk(rt|||}nY}wxYw)a Return the result of evaluating op on the passed in values. If native types are not compatible, try coercion to object dtype. Parameters ---------- left : np.ndarray right : np.ndarray or scalar Excludes DataFrame, Series, Index, ExtensionArray. is_cmp : bool, default False If this a comparison operation. Returns ------- array-like Raises ------ TypeError : invalid operation r7N)r3strrrevaluaterMr7objectgetattrrWrNotImplementedr#r!dispatch_fill_zeros)r)r*rBis_cmpfuncrEs r/_na_arithmetic_opras,%{++R0 dE"9V$.(@"$r22  & &r4 ??%   JJ& GE7D$AV$K &dE26F   s A22AB65B6ct|tr t|d}t|}t ||s#t|t t tfs|tur |||}|St|||t|||}|S)aO Evaluate an arithmetic operation `+`, `-`, `*`, `/`, `//`, `%`, `**`, ... Note: the caller is responsible for ensuring that numpy warnings are suppressed (with np.errstate(all="ignore")) if needed. Parameters ---------- left : np.ndarray or ExtensionArray right : object Cannot be a DataFrame or Index. Series is *not* excluded. op : {operator.add, operator.sub, ...} Or one of the reversed variants from roperator. Returns ------- ndarray or ExtensionArray Or a 2-tuple of these in the case of divmod or rdivmod. N) r3r4r rr"rr r r_bool_arith_checkra)r)r*rB res_valuess r/ arithmetic_opres2%ud+ *5 1E "$. eiY? @ C<e_   "dE*'tUB7 r1ct|}t|}tj|}t|dddk(}t |t r t |d}t|}t |tjtfr:|s8t|t|k7r!td|j|jt||s6t |ttt fs|t"ur|j$t&k7r |||}|St)|s|rkt+|r`|t,j.ur'tj0|jt2}|Stj4|jt2}|St7||r t9|||S|j$t&k(st |t:rt=|||}|St?|||d}|S)a Evaluate a comparison operation `=`, `!=`, `>=`, `>`, `<=`, or `<`. Note: the caller is responsible for ensuring that numpy warnings are suppressed (with np.errstate(all="ignore")) if needed. Parameters ---------- left : np.ndarray or ExtensionArray right : object Cannot be a DataFrame, Series, or Index. op : {operator.eq, operator.ne, operator.gt, operator.ge, operator.lt, operator.le} Returns ------- ndarray or ExtensionArray ndimNrzLengths must match to comparerHT)r_) rr item_from_zerodimr\r3r4r r5r6rrLr<r;r"rr r rr7r[rroperatorneonesboolzerosrr#rYrFra)r)r*rBlvaluesrvaluesrvalues_is_zerodimrds r/ comparison_oprq&s&-T2G,U3G##G,G 'w=B'4 !$/,W5G'BJJ(9:;DV w<3w< '/  !'2 GiY? @ESL MMV #) ( # G  2W  d;J '--t>!'' ""7  a $ + 0C DDa Aa A%%aggiB?F==# ##7G ,,Q26#  1g&&&r{{m3DQWWIN++.%q2 s4 &CE76D E7E0,?E++E00E76E7cdd}tj|}t|rt|ds t dt |}|}t ||r |||}|St|tjr$|jjdv}|s|||}ntj|}t|||}|jjdvr|s||}|S)ag Evaluate a logical operation `|`, `&`, or `^`. Parameters ---------- left : np.ndarray or ExtensionArray right : object Cannot be a DataFrame, Series, or Index. op : {operator.and_, operator.or_, operator.xor} Or one of the reversed variants from roperator. Returns ------- ndarray or ExtensionArray c|jjdvr5t|}|jr|j t }d||<||jjdk(r|j t }|S)NcfOFrs)r7rurr'r8r[rl)rCr)r.s r/ fill_boolzlogical_op..fill_boolsa 77<<5 7DxxzHHV$$ <4::??c1Ar1r7zLogical ops (and, or, xor) between Pandas objects and dtype-less sequences (e.g. list, tuple) are no longer supported. Wrap the object in a Series, Index, or np.array before operating instead.iu)N)r rhrhasattrrMrr"r3r5r6r7ru is_integerr~)r)r*rBrrnrordis_other_int_dtypes r/ logical_oprs"   ! !% (EE75'#: (  -T2GG '2) & ! grzz *!(!3!3t!; %#GW5"%!8 "7GR8  4',>":.J r1ct|tr|S|jjdj d}|dk(r|S|dvrtt |S|dvrtt |S|dvrtt|St|)z Return a binary array operation corresponding to the given operator op. Parameters ---------- op : function Binary operator from operator or roperator module. Returns ------- functools.partial _rarith_op>eqgegtleltrj)rB>orandrorxorrandrxor>addmodmulrOsubdivmodtruedivfloordiv) r3rr{striplstriprqrrerz)rBop_names r/ get_array_oprs"g kk$++C0G* 66},, ? ?zb))   },,!'**r1c^t|tjur t|St|tjur t |St |t jrt|rddl m }t|jr|jd}n;t|js&t|j}|j|}t j ||}|j#||jSt |St |t j$rt|rddl m}t|jr|jd}n;t|js&t|j}|j|}t j ||}|j#||jSt|St |t j(r t+|St |t j,r t/|S|S)ax Cast non-pandas objects to pandas types to unify behavior of arithmetic and comparison operations. Parameters ---------- obj: object shape : tuple[int] Returns ------- out : object Notes ----- Be careful to call this *after* determining the `name` attribute to be attached to the result of the arithmetic operation. r) DatetimeArrayz datetime64[s]rH)TimedeltaArrayztimedelta64[s])rNdatetime timedeltarr r3r5 datetime64rpandas.core.arraysrrr7r8rr broadcast_to _simple_new timedelta64rintegerintfloatingfloat)objr;r new_dtyper*rs r/maybe_prepare_scalar_for_oprs& CyH&&&~ ch'' '~ C ' 9 8399%jj1' 2/ : jj+OOC/E ,,U%++,F F~ C ( 9 9 399%jj!12' 2/ : jj+OOC/E!--e5;;-G G ~ C $3x C %Sz Jr1c|tvrt|jjdk(rZt|st j |r9|j jdjd}td|dyyy)z` In contrast to numpy, pandas raises an error for certain operations with booleans. rsrrz operator 'z!' not implemented for bool dtypesN) _BOOL_OP_NOT_ALLOWEDr7rurr is_boolr{rrrz)rBarsrs r/rcrcbsu  !! 77<<3 M!$4 Akk'',33C8G%WI%FG 9G "r1)rC np.ndarrayreturnr)F)r)rr_rl)r)r$r*r)r)r$r*rrr$)rCr)r;r%)rrrNone)M__doc__ __future__rr functoolsrritypingrrnumpyr5 pandas._libsrrr r r r=pandas._libs.tslibsr r rrpandas.core.dtypes.castrrpandas.core.dtypes.commonrrrrrrpandas.core.dtypes.genericrrrpandas.core.dtypes.missingrr pandas.corerpandas.core.computationrpandas.core.constructionrr pandas.core.opsr!pandas.core.ops.dispatchr"pandas.core.ops.invalidr#pandas._typingr$r%r0rFrWrarerqr~rrrrrtruedivr rfloordivrOrQrrdivmodrrcr1r/rs #   "/$>6&R#(5p0@f1hCL&#RBJ(+VNd     LL NN    r1