o ҷh{0@sddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZmZerXddlZdd lmZmZmZmZmZeGd d d eZ ddddZ!GdddeZ"dS)) annotationsN) TYPE_CHECKINGcast)libmissing) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeDtypeObjSelfnpttype_tc@s|eZdZdZdZedddZeddd Zedd d Ze d ddZ dddZ ed!ddZ ed!ddZ d"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 booleanreturntypecCstjSN)npbool_selfrM/var/www/html/venv/lib/python3.10/site-packages/pandas/core/arrays/boolean.pyrCzBooleanDtype.typestrcCdS)NbrrrrrkindGzBooleanDtype.kindnp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeKs zBooleanDtype.numpy_dtypetype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrrconstruct_array_typeOs z!BooleanDtype.construct_array_typecCr )Nrrrrrr__repr__ZszBooleanDtype.__repr__r%cCr NTrrrrr _is_boolean]r#zBooleanDtype._is_booleancCr r-rrrrr _is_numericar#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_buffersr4to_numpy null_countzerosappendr0_concat_same_type) rr0r6r=r>maskdataresultsarrbuflistbool_arrrrr__from_arrow__esF       zBooleanDtype.__from_arrow__N)rr)rr)rr$)rr(rr%)r0r1rr))__name__ __module__ __qualname____doc__namepropertyrr"r' classmethodr+r,r.r/rNrrrrr'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 inputiufcbr2zNeed to pass bool-like valuesTskipna)floatingintegerzmixed-integer-float)rr@npt.NDArray[np.bool_]z&values.shape and mask.shape must match)r:r) ValueError_data_maskrWrndarrayr&rr"r rEr<r%astypeallr9asarrayobjectr infer_dtyperfloatshaper0)valuesrHrW mask_values values_bool values_objectinferred_dtype integer_likerrrcoerce_to_arraysl    rpcseZdZdZdZdZdZhdZhdZe 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.00FALSEFalsefalserj np.ndarrayrHr^rrcst||}t|_|Sr)super _simple_newr_dtype)r*rjrHresult __class__rrr},szBooleanArray._simple_newrWr%Nonecs>t|tjr |jtjkstdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrW) r:rrbr&rr9rr~r|__init__)rrjrHrWrrrr2s zBooleanArray.__init__rcCs|jSr)r~rrrrr&=rzBooleanArray.dtypeN)r&rW 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)r_)sfalse_values_uniontrue_values_unionrr map_stringNs z:BooleanArray._from_sequence_of_strings..map_stringr2)r&rWrO) _TRUE_VALUESunion _FALSE_VALUESrr0rfr listmap_from_sequence) r*rr&rWrrrscalarsrHrrr_from_sequence_of_stringsAs z&BooleanArray._from_sequence_of_stringsrrrXcCs|r|dksJt||dS)Nrr)rp)r*valuer&rWrrr_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%r2r3z(can only perform ops with 1-d structuresFrz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rPr is_scalarr:r)r`rarrrendimNotImplementedErrorrpritem libmissingNAis_boolr9rr<r_r kleene_or kleene_and kleene_xor_maybe_mask_result)rotheropother_is_scalarrHrrrr_logical_methodes4       zBooleanArray._logical_methodrZrTrr[r cKsr|j}|j}|dvr"tt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|S)N)cummincummaxr[r) IntegerArray) r`ragetattrr r}pandas.core.arraysrrcint _accumulate)rrTr[kwargsrIrHrrrrrrs   zBooleanArray._accumulate)rjr{rHr^rr)F)rjr{rHr{rWr%rr)rr) rrr&rrWr%rrrrrr))r&rrWr%rrX)rTrr[r%rr )rPrQrRrS_internal_fill_value _truthy_value _falsey_valuerrrVr}rrUr&rrrbnumbersNumberr%r_HANDLED_TYPESrrr __classcell__rrrrr)s44 $r))NF)rWr%rrX)# __future__rrtypingrrnumpyr pandas._libsrrrpandas.core.dtypes.commonrpandas.core.dtypes.dtypesrpandas.core.dtypes.missingr pandas.corer pandas.core.array_algosr pandas.core.arrays.maskedr r r6pandas._typingrrrrrrrpr)rrrrs&       p W