o h0@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_selfrL/var/www/html/env/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)zero_copy_only)pyarrowrrtypesis_null TypeError isinstanceArraylenchunkslengthronesr'emptyr+buffers from_buffersr6to_numpy null_countzerosappendr2_concat_same_type) rr2r8r?r@maskdataresultsarrbuflistbool_arrrrr__from_arrow__fsF       zBooleanDtype.__from_arrow__N)rr)rr!)rr&)rr*rr')r2r3rr+)__name__ __module__ __qualname____doc__r__annotations__propertyrr$r) classmethodr-r.r0r1rPrrrrr(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 inputiufcbr4zNeed to pass bool-like valuesTskipna)floatingintegerzmixed-integer-float)rrBnpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r<r+ ValueError_data_maskrYrndarrayr(rr$r rGr>r'astypeallr;asarrayobjectr infer_dtyperfloatshaper2)valuesrJrY mask_values values_bool values_objectinferred_dtype integer_likerrrcoerce_to_arraysl    rrcseZdZdZdZdZdZhdZhdZe 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>1.01TRUETruetrue>0.00FALSEFalsefalserl np.ndarrayrJr`rrcst||}t|_|Sr)super _simple_newr_dtype)r,rlrJresult __class__rrr-szBooleanArray._simple_newrYr'Nonecs>t|tjr |jtjkstdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrY) r<rrdr(rr;rrr~__init__)rrlrJrYrrrr3s zBooleanArray.__init__rcCs|jSr)rrrrrr(>r zBooleanArray.dtypeN)r(rY true_values false_valuesstrings list[str]r( Dtype | Nonerlist[str] | Nonerc sp|j|pg|j|pgdfdd }tj|td}t|}tt|||||<|j |||dS)Nrr'cs&|vrdS|vr dSt|d)NTFz cannot be cast to bool)ra)sfalse_values_uniontrue_values_unionrr map_stringOs z:BooleanArray._from_sequence_of_strings..map_stringr4)r(rYrQ) _TRUE_VALUESunion _FALSE_VALUESrr2rhr listmap_from_sequence) r,rr(rYrrrscalarsrJrrr_from_sequence_of_stringsBs z&BooleanArray._from_sequence_of_stringsrrrZcCs|r|dksJt||dS)Nrr)rr)r,valuer(rYrrr_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>or_xorand_ror_rxorrand_r'r4r5z(can only perform ops with 1-d structuresFrz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rRr is_scalarr<r+rbrcrrrgndimNotImplementedErrorrrritem libmissingNAis_boolr;rr>rar kleene_or kleene_and kleene_xor_maybe_mask_result)rotheropother_is_scalarrJrrrr_logical_methodfs4       zBooleanArray._logical_methodr\rr!r]r cKsr|j}|j}|dvr"tt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|S)N)cummincummaxr]r) IntegerArray) rbrcgetattrr rpandas.core.arraysrreint _accumulate)rrr]kwargsrKrJrrrrrrs   zBooleanArray._accumulate)rlr}rJr`rr)F)rlr}rJr}rYr'rr)rr) rrr(rrYr'rrrrrr+)r(rrYr'rrZ)rr!r]r'rr )rRrSrTrU_internal_fill_value _truthy_value _falsey_valuerrrXrrrWr(rrrdnumbersNumberr'r_HANDLED_TYPESrrr __classcell__rrrrr+s44 $r+)NF)rYr'rrZ)$ __future__rrtypingrrrnumpyr pandas._libsrrrpandas.core.dtypes.commonrpandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corer pandas.core.array_algosr pandas.core.arrays.maskedr rr8pandas._typingrrrrrrrrr+rrrrs&       p W