U c @s4UddlmZddlZddlmZddlmZddlm Z m Z Gddde Z Gdd d e Z d Z eGd d d e ZeGd dde ZeGddde ZeGddde ZeGddde ZeGddde ZeGddde ZeGddde ZeeeeeeeedZded<dS)) annotationsN)register_extension_dtype)is_integer_dtype) NumericArray NumericDtypec@sVeZdZdZeejZeZ e ddddZ e ddZ e dd d dd d d Z 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. ztype[IntegerArray])returncCstS)zq Return the array type associated with this dtype. Returns ------- type ) IntegerArrayclsr >/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/arrays/integer.pyconstruct_array_types z!IntegerDtype.construct_array_typecCstS)N)INT_STR_TO_DTYPEr r r r _str_to_dtype_mapping&sz"IntegerDtype._str_to_dtype_mappingz np.ndarrayznp.dtypebool)valuesdtypecopyrc Cs~z|j|d|dWStk rx}zH|j||d}||krJ|WY$Std|jdt||W5d}~XYnXdS)z Safely cast the values to the given dtype. "safe" in this context means the casting is lossless. e.g. if 'values' has a floating dtype, each value must be an integer. safe)Zcastingr)rz"cannot safely cast non-equivalent z to N)Zastype TypeErrorallrnp)r rrrerrZcastedr r r _safe_cast*s  zIntegerDtype._safe_castN)__name__ __module__ __qualname____doc__rrint64Z_default_np_dtyperZ_checker classmethodrrrr r r r rs   rc@s eZdZdZeZdZdZdZdS)r a> 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 rN) rrrrrZ _dtype_clsZ_internal_fill_valueZ _truthy_valueZ _falsey_valuer r r r r >s Cr z An ExtensionDtype for {dtype} integer data. .. versionchanged:: 1.0.0 Now uses :attr:`pandas.NA` as its missing value, rather than :attr:`numpy.nan`. Attributes ---------- None Methods ------- None c@s"eZdZejZdZejddZ dS) Int8DtypeZInt8int8rN) rrrrr#typename_dtype_docstringformatrr r r r r"sr"c@s"eZdZejZdZejddZ dS) Int16DtypeZInt16int16r$N) rrrrr*r%r&r'r(rr r r r r)sr)c@s"eZdZejZdZejddZ dS) Int32DtypeZInt32int32r$N) rrrrr,r%r&r'r(rr r r r r+sr+c@s"eZdZejZdZejddZ dS) Int64DtypeZInt64rr$N) rrrrrr%r&r'r(rr r r r r-sr-c@s"eZdZejZdZejddZ dS) UInt8DtypeZUInt8uint8r$N) rrrrr/r%r&r'r(rr r r r r.sr.c@s"eZdZejZdZejddZ dS) UInt16DtypeZUInt16uint16r$N) rrrrr1r%r&r'r(rr r r r r0sr0c@s"eZdZejZdZejddZ dS) UInt32DtypeZUInt32uint32r$N) rrrrr3r%r&r'r(rr r r r r2sr2c@s"eZdZejZdZejddZ dS) UInt64DtypeZUInt64uint64r$N) rrrrr5r%r&r'r(rr r r r r4sr4)r#r*r,rr/r1r3r5zdict[str, IntegerDtype]r) __future__rZnumpyrZpandas.core.dtypes.baserZpandas.core.dtypes.commonrZpandas.core.arrays.numericrrrr r'r"r)r+r-r.r0r2r4r__annotations__r r r r sB   0M