U
|
¬ý°daF ã @ sÌ d Z ddlmZ ddlZddlmZ ddlZddlmZ ddl Z
|
ddl m Z m Z mZmZmZ ddlmZmZmZmZmZmZ ddlmZmZ dd l m Z m Z dd
|
l m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& dd l'm(Z(m)Z)m*Z* dd l+m,Z,m-Z- dd l.m/Z/ ddl0m1Z1 ddl2m3Z3m4Z4 ddl5m6Z6 ddl7m8Z8 dd Z9ddddZ:d1ddddd Z;d d d d d!Z<d d d d"d#d$Z=ddd%d&Z>d d d d"d'd(Z?d)d* Z@d+d,d-d.ZAejBe4jCejDe4jEejFe4jGhZHd/d0 ZIdS )2zY
|
Functions for arithmetic and comparison operations on NumPy arrays and
|
ExtensionArrays.
|
é )Ú annotationsN)Úpartial)ÚAny)ÚNaTÚ TimedeltaÚ TimestampÚlibÚops)Ú
|
BaseOffsetÚget_supported_resoÚget_unit_from_dtypeÚis_supported_unitÚ is_unitlessÚnpy_unit_to_abbrev)Ú ArrayLikeÚShape)Ú'construct_1d_object_array_from_listlikeÚfind_common_type)Ú ensure_objectÚ is_bool_dtypeÚis_integer_dtypeÚ is_list_likeÚis_numeric_v_string_likeÚis_object_dtypeÚ is_scalar)ÚABCExtensionArrayÚABCIndexÚ ABCSeries)ÚisnaÚnotna)Ú expressions)Ú ensure_wrapped_if_datetimelike)ÚmissingÚ roperator)Úshould_extension_dispatch)Úinvalid_comparisonc C s t |trt|}t |tjttfrt|js:| tj
|
¡}t |ttfrN|j }|j |j krjt d|j |j t | ¡ | ¡ | ¡}nt | ¡ || ¡}| |j ¡S )NzShapes must match)Ú
|
isinstanceÚlistr ÚnpÚndarrayr r r ÚdtypeÚastypeZobject_Z_valuesÚshapeÚ
|
ValueErrorÚlibopsZ vec_compareÚravelZscalar_compareÚreshape)ÚopÚxÚyÚresult© r5 úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/ops/array_ops.pyÚcomp_method_OBJECT_ARRAYC s
|
|
r7 z
|
np.ndarray©r2 c C sV | ¡ }t| tjs tt| t|tjrªt| j|jg}tj| j |d}t
|
| t
|
|krjt | j |j t |}| ¡ }t || ¡ @ }| ¡ r¨||| || ||< nt|sÄtdt| tj| j | jd}t |}|tkrút | dkd|¡}n |tjkrt |dkd|¡}| ¡ r4||| |||< t || tj¡ | | 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
|
©r* z1Cannot broadcast np.ndarray with operand of type é F)r/ r& r( r) ÚAssertionErrorÚtyper r* ÚemptyÚsizeÚlenr- r, r Úanyr Ú TypeErrorÚpowÚwherer# ÚrpowZputmaskÚnanr0 ) r2 r3 r1 Zxravr* r4 ZymaskZyravÚmaskr5 r5 r6 Ú_masked_arith_opV s6 ÿ
|
rG FÚbool)ÚleftÚis_cmpc C s t |tr|}n ttj|}z|| |}W n: tk
|
rd |s^t| jsPt|r^t| ||}n Y nX |rt |sz|t
|
krt | ||S t || ||¡S )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
|
)r& Ústrr r ÚevaluaterA r r* rG r ÚNotImplementedr% r" Zdispatch_fill_zeros)rI Úrightr1 rJ Úfuncr4 r5 r5 r6 Ú_na_arithmetic_op s
|
rP r r )rI rN c C sJ t | |s"t|tttfs"|tkr.|| |}nt|| | 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.
|
)r$ r& r r
|
r r Ú_bool_arith_checkrP )rI rN r1 Ú
|
res_valuesr5 r5 r6 Ú arithmetic_op¿ s ÿþý rS )rI rN Úreturnc C s$ t | }t |}t |¡}t|tr.t |¡}t|tjtfr^t |t |kr^t
|
d|j |j t ||st|t ttfs|tkrt|js|||}nt|rÔt|rÔ|tjkrÂtj|j td}ntj|j td}nLt||rêt|||S t|jst|t rt |||}nt |||dd}|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
|
z Lengths must match to comparer9 T)rJ ) r! r Úitem_from_zerodimr& r' r( Zasarrayr) r r? r- r, r$ r r
|
r r r r* r r ÚoperatorÚneZonesrH Úzerosr r% rK r7 rP )rI rN r1 ÚlvaluesÚrvaluesrR r5 r5 r6 Ú comparison_opí s<
|
|
|
ÿ
|
ÿÿþ
|
|
r[ c C s z|| |}W nê t k
|
rø t|tjrjt| jrBt|jrBtt| } t|}t |
|
¡ |
|
¡ |¡}nt |¡sxtt |st|}zt | ||¡}W nX t ttttfk
|
rò } z0t|j}t d|j d| j d| d|W 5 d }~X Y nX Y nX | | j¡S )NzCannot perform 'z' with a dtyped [z ] array and scalar of type [ú])rA r& r( r) r r* r; r r. Z vec_binopr/ r r r rH Z scalar_binopr- ÚAttributeErrorÚ OverflowErrorÚNotImplementedErrorr< Ú__name__r0 r, )r2 r3 r1 r4 ÚerrÚtypr5 r5 r6 Ú na_logical_op- s6 û
|
ÿýrc c C s¾ dd }ddd}t | j}t |¡}t|r@t|ds@t|}t| }|}t||rb|||}nXt |t
|
j rt |j} | r|n|||}n
|
t |¡} |r¢| r¢|n|}
|
t |||}|
|
|}|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 S s | S )Nr5 r8 r5 r5 r6 Ú<lambda>f ó z logical_op.<locals>.<lambda>Nc S sN | j jdkr.t| }| ¡ r.| t¡} d| |< |d ks@t|j rJ| t¡} | S )N)ÚcÚfÚOF)r* Úkindr r@ r+ Úobjectr rH )r2 rI rF r5 r5 r6 Ú fill_boolh s
|
|
z logical_op.<locals>.fill_boolr* )N)r r* r rU r Úhasattrr r! r$ r& r( r) Ú
|
is_integerrc ) rI rN r1 Zfill_intrk Zis_self_int_dtyperY rZ rR Zis_other_int_dtypeZfillerr5 r5 r6 Ú
|
logical_opV s$
|
|
|
|
|
|
rn c C st t | tr| S | j d¡ d¡}|dkr,| S |dkr@tt| dS |dkrTtt| dS |dkrhtt| dS t|dS ) 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
|
Ú_ÚrZarith_op> ÚleÚltrW ÚgeÚgtÚeq)r1 > ÚxorÚrandÚrxorÚorÚrorÚand> ÚaddÚtruedivÚdivmodÚfloordivÚmodÚsubrB ÚmulN) r& r r` ÚstripÚlstripr[ rn rS r_ )r1 Úop_namer5 r5 r6 Ú get_array_op s
|
|
r r )r, c C sF t | tjkrt| S t | tjkr,t| S t| tjr´t| r¬ddl m
|
} t | j rb| d¡} n6tt| j st| j }tt|}| d| d¡} t | |¡}||S t| S t| tjrBt| r:ddl m} t | j rî| d¡} n8tt| j s&t| j }tt|}| d| d¡} t | |¡}||S 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 )Ú DatetimeArrayzdatetime64[ns]z datetime64[r\ )ÚTimedeltaArrayztimedelta64[ns]z timedelta64[)r< ÚdatetimeÚ timedeltar r r& r( Z
|
datetime64r Zpandas.core.arraysr r r* r+ r r r r Z broadcast_toZ timedelta64r )Úobjr, r ÚunitZ closest_unitrN r r5 r5 r6 Úmaybe_prepare_scalar_for_opà s:
|
|
|
|
|
r c C sP | t krLt|jrLt|s*t|ttjfrL| j d¡ d¡}t
|
d| ddS )z`
|
In contrast to numpy, pandas raises an error for certain operations
|
with booleans.
|
ro rp z
|
operator 'z!' not implemented for bool dtypesN) Ú_BOOL_OP_NOT_ALLOWEDr r* r& rH r( Zbool_r` r r r_ )r1 ÚaÚbr
r5 r5 r6 rQ s
|
ÿÿ
|
ÿrQ )F)JÚ__doc__Ú
|
__future__r r Ú functoolsr rV Útypingr Únumpyr( Z pandas._libsr r r r r r. Zpandas._libs.tslibsr
|
r r r r r Zpandas._typingr r Zpandas.core.dtypes.castr r Zpandas.core.dtypes.commonr r r r r r r Zpandas.core.dtypes.genericr r r Zpandas.core.dtypes.missingr r Zpandas.core.computationr Zpandas.core.constructionr! Zpandas.core.opsr" r# Zpandas.core.ops.dispatchr$ Zpandas.core.ops.invalidr% r7 rG rP rS r[ rc rn r r r} Zrtruedivr Z rfloordivrB rD r rQ r5 r5 r5 r6 Ú<module> sF $ 81.@)B+Hú
|
