B W0d3@s8ddlmZddlZddlZddlmZmZddl m Z m Z ddl m ZddlmZddlmZddlmZmZmZmZmZmZmZdd lmZmZdd lmZdd l m!Z!m"Z"dd l#m$Z$dd l%m&Z&Gddde"Z'ddddddZ(Gddde!Z)dZ*eGddde'Z+eGddde'Z,e+e,dZ-dS)) annotationsN)libmissing) ArrayLikeDtypeObj)function)cache_readonly)astype_nansafe) is_bool_dtypeis_datetime64_dtypeis_float_dtypeis_integer_dtype is_list_likeis_object_dtype pandas_dtype)ExtensionDtyperegister_extension_dtype)isna) NumericArray NumericDtype)invalid_comparison) to_numericc@sReZdZdZddddZeddddZed dd d Zd d dddZ dS) FloatingDtypea An ExtensionDtype to hold a single size of floating dtype. These specific implementations are subclasses of the non-public FloatingDtype. For example we have Float32Dtype to represent float32. The attributes name & type are set when these subclasses are created. str)returncCs |jdS)NzDtype())name)selfrM/var/www/html/venv/lib/python3.7/site-packages/pandas/core/arrays/floating.py__repr__4szFloatingDtype.__repr__boolcCsdS)NTr)rrrr _is_numeric7szFloatingDtype._is_numericztype[FloatingArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) FloatingArray)clsrrrconstruct_array_type;s z"FloatingDtype.construct_array_typezlist[DtypeObj]zDtypeObj | None)dtypesrcCsJtdd|DsdStdd|Dg}t|tjrFtt|SdS)Ncss|]}t|tVqdS)N) isinstancer).0trrr Hsz2FloatingDtype._get_common_dtype..cSsg|] }|jqSr) numpy_dtype)r'r(rrr Msz3FloatingDtype._get_common_dtype..)allnpZfind_common_typeZ issubdtypefloatingFLOAT_STR_TO_DTYPEr)rr%Znp_dtyperrr_get_common_dtypeFs  zFloatingDtype._get_common_dtypeN) __name__ __module__ __qualname____doc__rpropertyr! classmethodr$r0rrrrr*s  rFr ztuple[np.ndarray, np.ndarray])copyrc CsL|dkr"t|dr"t|jr"|j}|dk rt|trF|drF|}tt|t syt tt |}Wn2t k r}zt d||Wdd}~XYnXt|tr|j|j}}|dk r|j|jdd}|r|}|}||fSt j||d}t|rRtj|dd}|d kr6t t|}|t jn|d krt|jd nJt|rxt|rxt j|t|d }n$t|st|st|jd |dkrt |}nt|t|kst!|j"d kstd|j"d kstd|dkrt d}n|j}|#r6|}t j||<|j|dd}n|j|dd}||fS)a, Coerce the input values array to numpy arrays with a mask. Parameters ---------- values : 1D list-like dtype : float dtype mask : bool 1D array, optional copy : bool, default False if True, copy the input Returns ------- tuple of (values, mask) NdtypeFloatzinvalid dtype specified F)r7T)skipnaempty)r.integerz mixed-integerz integer-nazmixed-integer-floatz' cannot be converted to a FloatingDtype)r8r7zvalues must be a 1D list-likezmask must be a 1D list-likefloat64)$hasattrr r8r&r startswithlower issubclasstyperr/r-KeyError ValueErrorr"_data_maskastyper*r7arrayrrZ infer_dtyper;lenfillnan TypeErrorr floatr rAssertionErrorndimany)valuesr8maskr7errZ inferred_typerrrcoerce_to_arrayUs^ "            rUcseZdZdZdZeddddZd.ddd d fd d Zed ddd ddddZ ed ddd ddddZ ddddZ d/d ddfdd Z ddddZ ddZddd fd!d" Zddd fd#d$ Zdd%fd&d' Zdd%fd(d) Zd*d+d,d-ZZS)0r"a Array of floating (optional missing) values. .. versionadded:: 1.2.0 .. warning:: FloatingArray is currently experimental, and its API or internal implementation may change without warning. Especially the behaviour regarding NaN (distinct from NA missing values) is subject to change. We represent a FloatingArray with 2 numpy arrays: - data: contains a numpy float array of the appropriate dtype - mask: a boolean array holding a mask on the data, True is missing To construct an FloatingArray from generic array-like input, use :func:`pandas.array` with one of the float dtypes (see examples). See :ref:`integer_na` for more. Parameters ---------- values : numpy.ndarray A 1-d float-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 ------- FloatingArray Examples -------- Create an FloatingArray with :func:`pandas.array`: >>> pd.array([0.1, None, 0.3], dtype=pd.Float32Dtype()) [0.1, , 0.3] Length: 3, dtype: Float32 String aliases for the dtypes are also available. They are capitalized. >>> pd.array([0.1, None, 0.3], dtype="Float32") [0.1, , 0.3] Length: 3, dtype: Float32 gr)rcCstt|jjS)N)r/rrFr8)rrrrr8szFloatingArray.dtypeFz np.ndarrayr )rRrSr7cs6t|tjr|jjdks tdtj|||ddS)NfzJvalues should be floating numpy array. Use the 'pd.array' function instead)r7)r&r-Zndarrayr8kindrMsuper__init__)rrRrSr7) __class__rrrYszFloatingArray.__init__N)r8r7)r7rcCst|||d\}}t||S)N)r8r7)rUr")r#scalarsr8r7rRrSrrr_from_sequenceszFloatingArray._from_sequencecCst|dd}|j|||dS)Nraise)errors)r8r7)rr\)r#stringsr8r7r[rrr_from_sequence_of_stringss z'FloatingArray._from_sequence_of_stringsztuple[np.ndarray, np.ndarray]cCst||jdS)N)r8)rUr8)rvaluerrr_coerce_to_arrayszFloatingArray._coerce_to_arrayTrcstt|}t|tr"tj||dSt|r6dtji}nt|rNdt di}ni}|j fd|i|}t ||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, or BooleanArray, IntegerArray or FloatingArray with 'dtype' for its dtype. Raises ------ TypeError if incompatible type with an FloatingDtype, equivalent of same_kind casting )r7Zna_valueZNaTr8F) rr&rrXrHr r-rLr Z datetime64Zto_numpyr )rr8r7kwargsdata)rZrrrHs  zFloatingArray.astypecCs|jS)N)rF)rrrr_values_for_argsort@sz!FloatingArray._values_for_argsortc Cs ddlm}m}d}t|||tfr4|j|j}}n$t|rXt |}|j dkrXt d|t j krtj|jjdd}tj|jjdd}nnt`tddttjdd $t|jd |jd }||}WdQRX|tkrt|j||}WdQRX|dkr |j}n |j|B}|||S) Nr) BooleanArray IntegerArrayr=z(can only perform ops with 1-d structuresr )r8ignoreZ elementwise)r,__)Z pandas.arraysrfrgr&r"rFrGrr-ZasarrayrPNotImplementedError libmissingZNAZzerosshapeZoneswarningscatch_warningsfilterwarnings FutureWarningZerrstategetattrr1NotImplementedrr7)rotheroprfrgrSresultmethodrrr _cmp_methodCs,       zFloatingArray._cmp_methodr)r: min_countc std|tjd||dS)Nrsum)r:rx)nvZ validate_sumrX_reduce)rr:rxrc)rZrrryqs zFloatingArray.sumc std|tjd||dS)Nrprod)r:rx)rzZ validate_prodrXr{)rr:rxrc)rZrrr|us zFloatingArray.prod)r:c std|tjd|dS)Nrmin)r:)rzZ validate_minrXr{)rr:rc)rZrrr}ys zFloatingArray.minc std|tjd|dS)Nrmax)r:)rzZ validate_maxrXr{)rr:rc)rZrrr~}s zFloatingArray.maxr)op_namecCst|||ddS)z Parameters ---------- result : array-like mask : array-like bool other : scalar or array-like op_name : str F)r7)rC)rrurSrsrrrr_maybe_mask_resultsz FloatingArray._maybe_mask_result)F)T)r1r2r3r4Z_internal_fill_valuerr8rYr6r\r`rbrHrerwryr|r}r~r __classcell__rr)rZrr"s":..r"z An ExtensionDtype for {dtype} data. This dtype uses ``pd.NA`` as missing value indicator. Attributes ---------- None Methods ------- None c@s"eZdZejZdZejddZ dS) Float32DtypeZFloat32float32)r8N) r1r2r3r-rrCr_dtype_docstringformatr4rrrrrsrc@s"eZdZejZdZejddZ dS) Float64DtypeZFloat64r>)r8N) r1r2r3r-r>rCrrrr4rrrrrsr)rr>)NNF). __future__rrmnumpyr-Z pandas._libsrrrkZpandas._typingrrZpandas.compat.numpyrrzZpandas.util._decoratorsrZpandas.core.dtypes.castr Zpandas.core.dtypes.commonr r r r rrrZpandas.core.dtypes.dtypesrrZpandas.core.dtypes.missingrZpandas.core.arrays.numericrrZpandas.core.opsrZpandas.core.tools.numericrrrUr"rrrr/rrrrs0    $    ,`n