U f/e@sddlmZddlZddlZddlmZmZmZddlZ ddl m Z m Z ddlmZddlmZddlmZmZmZmZmZddlmZdd lmZmZerddlZed d d ZGd ddeZ Gdd d eZ!dS)) annotationsN) TYPE_CHECKINGAnyTypeVar) Timedeltamissing)functionAbstractMethodError)is_floatis_float_dtype is_integeris_integer_dtype is_list_like)ops)BaseMaskedArrayBaseMaskedDtypeT NumericArray)boundc@seZdZdddddZdS) NumericDtypez$pyarrow.Array | pyarrow.ChunkedArrayr)arrayreturnc Csddl}ddlm}|}||j}|j|s>||}t||j rR|g}n|j }g}|D]4}|||jd\} } || | dd} | | q`|s|t jg|jdt jgt jdSt|dkr|dS||SdS)zW Construct IntegerArray/FloatingArray from pyarrow Array/ChunkedArray. rN)pyarrow_array_to_numpy_and_maskdtypeF)copy)pyarrowZpandas.core.arrays._arrow_utilsrZconstruct_array_typeZfrom_numpy_dtypetypeequalscast isinstanceArraychunksrappendnprZ numpy_dtypeZbool_lenZ_concat_same_type) selfrrrZ array_classZ pyarrow_typer$resultsZarrdatamaskZnum_arrr,>/tmp/pip-unpacked-wheel-tiezk1ph/pandas/core/arrays/numeric.py__from_arrow__)s,       zNumericDtype.__from_arrow__N)__name__ __module__ __qualname__r.r,r,r,r-r(src@sneZdZdZddddZddZejej fZ ddd d d Z d d Z ddZ ddZdddddddZdS)rz8 Base class for IntegerArray and FloatingArray. str)op_namecCs t|dSNr )r(resultr+otherr3r,r,r-_maybe_mask_resultUszNumericArray._maybe_mask_resultc CsR|j}d}t|dddkr"tdt|tr<|j|j}}nt|rt |}|j dkr`tdt |t |krxt dt |st|stdn>t|tjtjfrt|}n"t|st|s|tjkstd|dkr|j}|tjkr|dO}n |j|B}|dkrjt|jdk|j@d |}|dk rJt|dk|@d |}n|tjk rt|dkd |}nh|d kr|dk rt|dk|@d |}n|tjk rt|dkd |}t|jdk|j@d |}|tjkrt|j}n$tjd d ||j|}W5QRX|d krB|\}}||||d||||dfS|||||S)Nndimrrz(can only perform ops with 1-d structureszLengths must matchz(can only perform ops with numeric valuesTpowFZrpowignore)alldivmodfloordivmod)r/getattrNotImplementedErrorr"r_data_maskrr&Zasarrayr8r' ValueErrorr r TypeErrordatetime timedeltaZ timedelta64rr r libmissingZNArwhereZ ones_likeZerrstater7) r(r6opr3Zomaskr+r5divr>r,r,r- _arith_methodXsZ                 zNumericArray._arith_methodznp.ufunc)ufuncmethodc s|dkrtd|dd}||D]}t||jtfs$tSq$tj|||f||}|tk rf|Stj t |t dg}|D]0}t|tr|j O| |jq| |qfddt||||}t|trtfdd |DS|SdS) Nreducez%The 'reduce' method is not supported.outr,rcs^t|jr(ddlm}}|||St|jrPddlm}}|||Stj|<|S)Nr) IntegerArray) FloatingArray) rrZpandas.core.arraysrPrr rQr&nan)xrPmrQ)r+r,r- reconstructs       z1NumericArray.__array_ufunc__..reconstructc3s|]}|VqdSr4r,).0rS)rUr,r- sz/NumericArray.__array_ufunc__..)r@getr"_HANDLED_TYPESrNotImplementedrZ!maybe_dispatch_ufunc_to_dunder_opr&zerosr'boolrBr%rAr?tuple) r(rLrMinputskwargsrOrSr5Zinputs2r,)r+rUr-__array_ufunc__s:        zNumericArray.__array_ufunc__cCst||j |jSr4)rrArBrr(r,r,r-__neg__szNumericArray.__neg__cCs|Sr4r,rar,r,r-__pos__szNumericArray.__pos__cCst|t|j|jSr4)rabsrArBrrar,r,r-__abs__szNumericArray.__abs__rrint)r(decimalsrcOs8t||tj|jfd|i|}t|||jS)a Round each value in the array a to the given number of decimals. Parameters ---------- decimals : int, default 0 Number of decimal places to round to. If decimals is negative, it specifies the number of positions to the left of the decimal point. *args, **kwargs Additional arguments and keywords have no effect but might be accepted for compatibility with NumPy. Returns ------- NumericArray Rounded values of the NumericArray. See Also -------- numpy.around : Round values of an np.array. DataFrame.round : Round values of a DataFrame. Series.round : Round values of a Series. rg)nvZvalidate_roundr&roundrArrBr)r(rgargsr_valuesr,r,r-ris zNumericArray.roundN)r)r/r0r1__doc__r7rKr&ZndarraynumbersNumberrYr`rbrcrerir,r,r,r-rPsC 4)" __future__rrErmtypingrrrZnumpyr&Z pandas._libsrrrGZpandas.compat.numpyrrhZ pandas.errorsr Zpandas.core.dtypes.commonr r r rrZ pandas.corerZpandas.core.arrays.maskedrrrrrrr,r,r,r-s     (