U f/e@ @sUddlmZddlZddlZddlmZmZmZ ddl m Z m Z m Z ddlmZddlmZddlmZmZddlmZmZmZmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&dd l'm(Z(dd l)m*Z*Gddde&Z+ddddZ,d/dddddZ-Gddde%Z.dZ/eGddde+Z0eGddde+Z1eGd d!d!e+Z2eGd"d#d#e+Z3eGd$d%d%e+Z4eGd&d'd'e+Z5eGd(d)d)e+Z6eGd*d+d+e+Z7e0e1e2e3e4e5e6e7d,Z8d-e9d.<dS)0) annotationsN)iNaTlibmissing) ArrayLikeDtypeDtypeObj)function)cache_readonly)ExtensionDtyperegister_extension_dtype) is_bool_dtypeis_datetime64_dtypeis_floatis_float_dtypeis_integer_dtype is_list_likeis_object_dtype pandas_dtype)isna)BaseMaskedArrayBaseMaskedDtype) NumericArray NumericDtype)invalid_comparison) to_numericc@sveZdZdZddddZeddddZeddd d Zeddd d Z e d dddZ dddddZ dS) _IntegerDtypea' An ExtensionDtype to hold a single size & kind of integer dtype. These specific implementations are subclasses of the non-public _IntegerDtype. For example we have Int8Dtype to represent signed int 8s. The attributes name & type are set when these subclasses are created. strreturncCs$|jr dnd}|dd|jdS)NUIntzDtype())is_unsigned_integeritemsize)selfsignr(>/tmp/pip-unpacked-wheel-tiezk1ph/pandas/core/arrays/integer.py__repr__:sz_IntegerDtype.__repr__boolcCs |jdkS)Nikindr&r(r(r)is_signed_integer>sz_IntegerDtype.is_signed_integercCs |jdkS)Nur-r/r(r(r)r$Bsz!_IntegerDtype.is_unsigned_integercCsdS)NTr(r/r(r(r) _is_numericFsz_IntegerDtype._is_numericztype[IntegerArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) IntegerArray)clsr(r(r)construct_array_typeJs z"_IntegerDtype.construct_array_typezlist[DtypeObj]zDtypeObj | None)dtypesrcCsptdd|DsdStdd|Dg}t|tjrFtt|St|tjrlddlm }|t|SdS)Ncss@|]8}t|tp6t|tjo6t|tjp6t|tjVqdSN) isinstancernpdtype issubdtypenumberZbool_.0tr(r(r) Ws  z2_IntegerDtype._get_common_dtype..cSs g|]}t|tr|jn|qSr()r8r numpy_dtyper=r(r(r) esz3_IntegerDtype._get_common_dtype..r)FLOAT_STR_TO_DTYPE) allr9Zfind_common_typer;integerINT_STR_TO_DTYPErfloatingZpandas.core.arrays.floatingrC)r&r6Znp_dtyperCr(r(r)_get_common_dtypeUs     z_IntegerDtype._get_common_dtypeN) __name__ __module__ __qualname____doc__r*r r0r$propertyr2 classmethodr5rHr(r(r(r)r0s  rr+copyc Cs~z|j|d|dWStk rx}zH|j||d}||krJ|WY$Std|jdt||W5d}~XYnXdS)z} Safely cast the values to the dtype if they are equivalent, meaning floats must be equivalent to the ints. safe)ZcastingrPrOz"cannot safely cast non-equivalent z to N)astype TypeErrorrDr:r9)valuesr:rPerrZcastedr(r(r) safe_castvs  rVFtuple[np.ndarray, np.ndarray]rPrc CsT|dkr"t|dr"t|jr"|j}|dk rt|trP|dsH|drP|}tt|t szt tt |}Wn2t k r}zt d||W5d}~XYnXt|tr|j|j}}|dk r|j|jdd}|r|}|}||fSt j||d}t|r\tj|dd }|d kr@t t|}|t jn|d krt|jd nJt|rt|rt j|t|d }n$t|st|st|jd |dkrt |}nt|t|kst!|j"dkstd|j"dkstd|dkrt d}n|j}|#r>|}d||<t$||dd}nt$||dd}||fS)a- Coerce the input values array to numpy arrays with a mask Parameters ---------- values : 1D list-like dtype : integer dtype mask : bool 1D array, optional copy : bool, default False if True, copy the input Returns ------- tuple of (values, mask) Nr:r"ZUIntzinvalid dtype specified FrOTskipnaempty)rGrEz mixed-integerz integer-nazmixed-integer-floatz' cannot be converted to an IntegerDtyper:rPzvalues must be a 1D list-likezmask must be a 1D list-likeint64)%hasattrrr:r8r startswithlower issubclasstyperrFr9KeyError ValueErrorr3_data_maskrRrArParrayrrZ infer_dtyper[lenfillnanrSr intrrAssertionErrorndimanyrV)rTr:maskrPrUZ inferred_typer(r(r)coerce_to_arraysf  "           rqcseZdZdZdZeddddZd0ddd d fd d Zed dddd ddddZ ed dddd ddddZ ddddZ d1d ddfdd Z ddddZ dd Zdd!d"fd#d$ Zdd!d"fd%d& Zdd'fd(d) Zdd'fd*d+ Zd,d-d.d/ZZS)2r3a> Array of integer (optional missing) values. .. versionchanged:: 1.0.0 Now uses :attr:`pandas.NA` as the missing value rather than :attr:`numpy.nan`. .. warning:: IntegerArray is currently experimental, and its API or internal implementation may change without warning. We represent an IntegerArray with 2 numpy arrays: - data: contains a numpy integer array of the appropriate dtype - mask: a boolean array holding a mask on the data, True is missing To construct an IntegerArray from generic array-like input, use :func:`pandas.array` with one of the integer dtypes (see examples). See :ref:`integer_na` for more. Parameters ---------- values : numpy.ndarray A 1-d integer-dtype array. mask : numpy.ndarray A 1-d boolean-dtype array indicating missing values. copy : bool, default False Whether to copy the `values` and `mask`. Attributes ---------- None Methods ------- None Returns ------- IntegerArray Examples -------- Create an IntegerArray with :func:`pandas.array`. >>> int_array = pd.array([1, None, 3], dtype=pd.Int32Dtype()) >>> int_array [1, , 3] Length: 3, dtype: Int32 String aliases for the dtypes are also available. They are capitalized. >>> pd.array([1, None, 3], dtype='Int32') [1, , 3] Length: 3, dtype: Int32 >>> pd.array([1, None, 3], dtype='UInt16') [1, , 3] Length: 3, dtype: UInt16 r]rrcCstt|jjSr7)rFrrfr:r/r(r(r)r:3szIntegerArray.dtypeFz np.ndarrayr+)rTrprPcs6t|tjr|jjdks tdtj|||ddS)N)r,r1zIvalues should be integer numpy array. Use the 'pd.array' function insteadrO)r8r9Zndarrayr:r.rSsuper__init__)r&rTrprP __class__r(r)rs7s zIntegerArray.__init__Nr\z Dtype | None)r:rPrcCst|||d\}}t||S)Nr\)rqr3)r4scalarsr:rPrTrpr(r(r)_from_sequence?szIntegerArray._from_sequencecCst|dd}|j|||dS)Nraise)errorsr\)rrw)r4stringsr:rPrvr(r(r)_from_sequence_of_stringsFs z&IntegerArray._from_sequence_of_stringsrWcCst||jdS)Nr:)rqr:)r&valuer(r(r)_coerce_to_arrayMszIntegerArray._coerce_to_arrayTrrXcs\t|}t|tr"tj||dSt|r2tj}nt|rFt d}nt j }|j ||ddS)a Cast to a NumPy array or ExtensionArray with 'dtype'. Parameters ---------- dtype : str or dtype Typecode or data-type to which the array is cast. copy : bool, default True Whether to copy the data, even if not necessary. If False, a copy is made only if the old dtype does not match the new dtype. Returns ------- ndarray or ExtensionArray NumPy ndarray, BooleanArray or IntegerArray with 'dtype' for its dtype. Raises ------ TypeError if incompatible type with an IntegerDtype, equivalent of same_kind casting rOZNaTF)r:na_valuerP) rr8r rrrRrr9rkrZ datetime64rZ no_defaultZto_numpy)r&r:rPrrtr(r)rRPs  zIntegerArray.astypecCs*|j}|jr&|d||j<|S)a= Return values for sorting. Returns ------- ndarray The transformed values should maintain the ordering between values within the array. See Also -------- ExtensionArray.argsort : Return the indices that would sort this array. r])rfrPrgromin)r&datar(r(r)_values_for_argsortzs  z IntegerArray._values_for_argsortc Cs.ddlm}d}t|tr*|j|j}}ns^   (   Fb{