o $j0@sddlmZddlZddlmZmZmZddlZddl m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZmZerZddlZdd lmZmZmZmZm Z eGd d d eZ! ddddZ"GdddeZ#dS)) annotationsN) TYPE_CHECKINGClassVarcast)libmissing) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeDtypeObjSelfnpttype_tc@seZdZUdZdZded<edddZed d d Zed!d dZ e d"ddZ d ddZ ed#ddZ ed#ddZd$ddZdS)% BooleanDtypeaI Extension dtype for boolean data. .. warning:: BooleanDtype is considered experimental. The implementation and parts of the API may change without warning. Attributes ---------- None Methods ------- None Examples -------- >>> pd.BooleanDtype() BooleanDtype booleanz ClassVar[str]namereturntypecCstjSN)npbool_selfr[/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/pandas/core/arrays/boolean.pyrDzBooleanDtype.typestrcCdS)NbrrrrrkindHzBooleanDtype.kindnp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeLs zBooleanDtype.numpy_dtypetype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrrconstruct_array_typePs z!BooleanDtype.construct_array_typecCr")Nrrrrrr__repr__[szBooleanDtype.__repr__r'cCr"NTrrrrr _is_boolean^r%zBooleanDtype._is_booleancCr"r/rrrrr _is_numericbr%zBooleanDtype._is_numericarray$pyarrow.Array | pyarrow.ChunkedArrayr+c Csjddl}|j|kr|j|jstd|jdt||jr)|g}t|}n|j }| }|j|jrJt j |t d}t j|t d}t||Sg}|D]N}|} |jj|jt|d| dg|jdjdd }|jdkr|jj|jt|d| dg|jdjdd }|}n t jt|t d}t||} || qN|stt jgt jdt jgt jdSt|S) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z insteadr()offsetF)Zzero_copy_only)pyarrowrrtypesZis_null TypeError isinstanceArraylenchunkslengthrZonesr'emptyr+buffersZ from_buffersr6Zto_numpyZ null_countzerosappendr2Z_concat_same_type) rr2r7r=r>maskdataresultsZarrZbuflistZbool_arrrrr__from_arrow__fsF       zBooleanDtype.__from_arrow__N)rr)rr!)rr&)rr*rr')r2r3rr+)__name__ __module__ __qualname____doc__r__annotations__propertyrr$r) classmethodr-r.r0r1rFrrrrr(s"         rFcopyr'rtuple[np.ndarray, np.ndarray]cCs,t|tr"|dur td|j|j}}|r|}|}||fSd}t|tjr7|jtj kr7|r6|}nt|tjrs|jj dvrst |}tj t |td}||t||<t|||j||ksptd|}nPtj|td}tj|dd}d}|d |vrtdtd t |}tj t |td}||t||<||vrt||t||tkstd|dur|durtj |jtd}n3|dur|}n,t|tjr|jtj kr|dur||B}n|r|}ntj|td}|dur||B}|j|jkrtd ||fS) a Coerce the input values array to numpy arrays with a mask. Parameters ---------- values : 1D list-like mask : bool 1D array, optional copy : bool, default False if True, copy the input Returns ------- tuple of (values, mask) Nz'cannot pass mask for BooleanArray inputZiufcbr4zNeed to pass bool-like valuesTskipna)Zfloatingintegerzmixed-integer-float)rr?npt.NDArray[np.bool_]z&values.shape and mask.shape must match)r:r+ ValueError_data_maskrOrndarrayr(rr$r rAr<r'astypeallr9asarrayobjectrZ infer_dtyperfloatshaper2)valuesrCrOZ mask_valuesZ values_boolZ values_objectZinferred_dtypeZ integer_likerrrcoerce_to_arraysl    r`cseZdZdZdZdZdZhdZhdZe d/fd d Z d0d1fdd Z e d2ddZ e dddddd3dd ZejejeejfZe dd!d4d$d%Zd&d'Zdd(d5d-d.ZZS)6r+a= Array of boolean (True/False) data with missing values. This is a pandas Extension array for boolean data, under the hood represented by 2 numpy arrays: a boolean array with the data and a boolean array with the mask (True indicating missing). BooleanArray implements Kleene logic (sometimes called three-value logic) for logical operations. See :ref:`boolean.kleene` for more. To construct an BooleanArray from generic array-like input, use :func:`pandas.array` specifying ``dtype="boolean"`` (see examples below). .. warning:: BooleanArray is considered experimental. The implementation and parts of the API may change without warning. Parameters ---------- values : numpy.ndarray A 1-d boolean-dtype array with the data. mask : numpy.ndarray A 1-d boolean-dtype array indicating missing values (True indicates missing). copy : bool, default False Whether to copy the `values` and `mask` arrays. Attributes ---------- None Methods ------- None Returns ------- BooleanArray Examples -------- Create an BooleanArray with :func:`pandas.array`: >>> pd.array([True, False, None], dtype="boolean") [True, False, ] Length: 3, dtype: boolean FT>TrueTRUE1z1.0true>z0.0falseFalseFALSE0r_ np.ndarrayrCrTrrcst||}t|_|Sr)super _simple_newr_dtype)r,r_rCresult __class__rrrk-szBooleanArray._simple_newrOr'Nonecs>t|tjr |jtjkstdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrO) r:rrXr(rr9rrlrj__init__)rr_rCrOrnrrrr3s zBooleanArray.__init__rcCs|jSr)rlrrrrr(>r zBooleanArray.dtypeN)r(rO true_values false_valuesstrings list[str]r( Dtype | Nonerslist[str] | Nonertc sp|j|pg|j|pgdfdd }tj|td}t|}tt|||||<|j |||dS)Nrr'cs&|vrdS|vr dSt|d)NTFz cannot be cast to bool)rU)sZfalse_values_unionZtrue_values_unionrr map_stringOs z:BooleanArray._from_sequence_of_strings..map_stringr4)r(rOrG) _TRUE_VALUESunion _FALSE_VALUESrr2r\r listmapZ_from_sequence) r,rur(rOrsrtr{ZscalarsrCrrzr_from_sequence_of_stringsBs z&BooleanArray._from_sequence_of_stringsrqrrPcCs|r|dksJt||dS)Nrrq)r`)r,valuer(rOrrr_coerce_to_array^s  zBooleanArray._coerce_to_arraycCs:|jdvsJt|}d}t|tr|j|j}}n't|r8tj |dd}|j dkr/t dt |dd\}}n t|tj rB|}|rY|tjurYt|sYtdt|jd |sgt|t|krgtd |jd vryt|j||j|\}}n|jd vrt|j||j|\}}n t|j||j|\}}|||S) N>Zrxorand_ror_rand_or_xorr'r4r5z(can only perform ops with 1-d structuresFrqz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rHrZ is_scalarr:r+rVrWrrr[ndimNotImplementedErrorr`ritem libmissingZNAZis_boolr9rr<rUr Z kleene_orZ kleene_andZ kleene_xorZ_maybe_mask_result)rotheropZother_is_scalarrCrmrrr_logical_methodfs4       zBooleanArray._logical_methodrQrr!rRr cKsr|j}|j}|dvr"tt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|S)N)ZcumminZcummaxrRr) IntegerArray) rVrWgetattrr rkZpandas.core.arraysrrYint _accumulate)rrrRkwargsrDrCrrrrrrs   zBooleanArray._accumulate)r_rirCrTrr)F)r_rirCrirOr'rrp)rr) rurvr(rwrOr'rsrxrtrxrr+)r(rrOr'rrP)rr!rRr'rr )rHrIrJrKZ_internal_fill_valueZ _truthy_valueZ _falsey_valuer|r~rNrkrrrMr(rrrXnumbersNumberr'rZ_HANDLED_TYPESrrr __classcell__rrrnrr+s44 $r+)NF)rOr'rrP)$ __future__rrtypingrrrnumpyrZ pandas._libsrrrZpandas.core.dtypes.commonrZpandas.core.dtypes.dtypesr Zpandas.core.dtypes.missingr Z pandas.corer Zpandas.core.array_algosr Zpandas.core.arrays.maskedr rr7Zpandas._typingrrrrrrr`r+rrrrs&       p W