o 5c1+@sddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z mZmZddlmZmZddlmZddlmZdd lmZdd lmZmZerVddlZdd l mZeGd d d eZ ddddZGdddeZ dS)) annotationsN) TYPE_CHECKINGcast)libmissing)DtypeDtypeObjtype_t) is_list_likeis_numeric_dtype)register_extension_dtype)isna)ops)BaseMaskedArrayBaseMaskedDtype)nptc@s|eZdZdZdZedddZeddd Zedd d Ze d ddZ dddZ ed!ddZ ed!ddZ d"ddZdS)# BooleanDtypeaf Extension dtype for boolean data. .. versionadded:: 1.0.0 .. warning:: BooleanDtype is considered experimental. The implementation and parts of the API may change without warning. Attributes ---------- None Methods ------- None Examples -------- >>> pd.BooleanDtype() BooleanDtype booleanreturntypecCstjSN)npbool_selfrP/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/core/arrays/boolean.pyrFzBooleanDtype.typestrcCdS)NbrrrrrkindJzBooleanDtype.kindnp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeNs zBooleanDtype.numpy_dtypetype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrrconstruct_array_typeRs z!BooleanDtype.construct_array_typecCr)Nrrrrrr__repr__]szBooleanDtype.__repr__r$cCrNTrrrrr _is_boolean`r"zBooleanDtype._is_booleancCrr,rrrrr _is_numericdr"zBooleanDtype._is_numericarray$pyarrow.Array | pyarrow.ChunkedArrayr(c Csddl}|j|krtd|jdt||jr|g}n|j}g}|D]N}|}|jj |jt |d|dg|j dj dd}|j dkr`|jj |jt |d|dg|j dj dd}|}n tjt |td }t||} || q%|sttjgtjd tjgtjd St|S) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z instead)offsetF)zero_copy_onlyr%)pyarrowrr TypeError isinstanceArraychunksbuffersr( from_bufferslenr2to_numpy null_countrzerosr$appendr/_concat_same_type) rr/r5r9resultsarrbuflistdatamaskbool_arrrrr__from_arrow__hs:     zBooleanDtype.__from_arrow__N)rr)rr)rr#)rr')rr$)r/r0rr()__name__ __module__ __qualname____doc__namepropertyrr!r& classmethodr*r+r-r.rHrrrrr(s"       rFcopyr$rtuple[np.ndarray, np.ndarray]cCs*t|tr"|dur td|j|j}}|r|}|}||fSd}t|tjr7|jtj kr7|r6|}nt|tjrrt |jrrt |}tj t |td}||t||<t|||j||ksotd|}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 inputr4zNeed to pass bool-like valuesT)skipna)floatingintegerzmixed-integer-float)remptyznpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r7r( ValueError_data_maskrPrndarrayr%rr r r?r<r$astypeallr6asarrayobjectr infer_dtyperfloatshaper/)valuesrFrP mask_values values_bool values_objectinferred_dtype integer_likerrrcoerce_to_arraysl    rgcseZdZdZdZdZdZhdZhdZ d$d%fd d Z e d&ddZ e dddddd'ddZ ejejeejfZe ddd(d d!Zd"d#ZZS))r(aZ 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). .. versionadded:: 1.0.0 .. 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>1.01TRUETruetrue>0.00FALSEFalsefalsera np.ndarrayrFrPr$rNonecs>t|tjr |jtjkstdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrP) r7rrYr%rr6r_dtypesuper__init__)rrarFrP __class__rrrw's zBooleanArray.__init__rcCs|jSr)rurrrrr%2rzBooleanArray.dtypeN)r%rP true_values false_valuesstrings list[str]r% Dtype | Nonerzlist[str] | Noner{csP|j|pg|j|pgfddfdd|D}|j|||dS)Ncs2t|r|S|vr dS|vrdSt|d)NTFz cannot be cast to bool)r rV)s)false_values_uniontrue_values_unionrr map_stringCsz:BooleanArray._from_sequence_of_strings..map_stringcsg|]}|qSrr).0x)rrr Msz:BooleanArray._from_sequence_of_strings..)r%rP) _TRUE_VALUESunion _FALSE_VALUES_from_sequence)r)r|r%rPrzr{scalarsr)rrrr_from_sequence_of_strings6s   z&BooleanArray._from_sequence_of_stringsrtrrQcCs|r|dksJt||dS)Nrrt)rg)r)valuer%rPrrr_coerce_to_arrayRs  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>or_xorand_ror_rxorrand_r$r4r1z(can only perform ops with 1-d structuresFrtz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rIr is_scalarr7r(rWrXr rr\ndimNotImplementedErrorrgritem libmissingNAis_boolr6rr<rVr kleene_or kleene_and kleene_xor_maybe_mask_result)rotheropother_is_scalarrFresultrrr_logical_methodZs4       zBooleanArray._logical_method)F)rarrrFrrrPr$rrs)rr) r|r}r%r~rPr$rzrr{rrr()r%rrPr$rrQ)rIrJrKrL_internal_fill_value _truthy_value _falsey_valuerrrwrNr%rOrrrYnumbersNumberr$r_HANDLED_TYPESrr __classcell__rrrxrr(s,6 r()NF)rPr$rrQ)! __future__rrtypingrrnumpyr pandas._libsrrrpandas._typingrrr pandas.core.dtypes.commonr r pandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corerpandas.core.arrays.maskedrrr5rrrgr(rrrrs&     j W