a Of0@sddlmZddlZddlmZmZmZddlZddl m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZmZerddlZdd lmZmZmZmZm Z eGd d d eZ!ddddddZ"GdddeZ#dS)) annotationsN) TYPE_CHECKINGClassVarcast)libmissing) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeDtypeObjSelfnpttype_tc@seZdZUdZdZded<eddddZed dd d Zed dd dZ e ddddZ d dddZ eddddZ eddddZdddddZdS) 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]nametypereturncCstjSN)npbool_selfrS/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/arrays/boolean.pyrDszBooleanDtype.typestrcCsdS)Nbrrrrr kindHszBooleanDtype.kindznp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeLszBooleanDtype.numpy_dtypeztype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrr construct_array_typePs z!BooleanDtype.construct_array_typecCsdS)Nrrrrrr __repr__[szBooleanDtype.__repr__r$cCsdSNTrrrrr _is_boolean^szBooleanDtype._is_booleancCsdSr+rrrrr _is_numericbszBooleanDtype._is_numericz$pyarrow.Array | pyarrow.ChunkedArrayr')arrayrc Csrddl}|j|kr6|j|js6td|jdt||jrR|g}t|}n|j }| }|j|jrt j |t d}t j|t d}t||Sg}|D]}|} |jj|jt|d| dg|jdjdd }|jdkr|jj|jt|d| dg|jdjdd }|}nt jt|t d}t||} || q|sdtt jgt jdt jgt jdSt|SdS) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z insteadr%)offsetF)Zzero_copy_only)pyarrowrrtypesZis_null TypeError isinstanceZArraylenchunkslengthrZonesr$emptyr'buffersZ from_buffersr1Zto_numpyZ null_countzerosappendr.Z_concat_same_type) rr.r2r7r8maskdataresultsZarrZbuflistZbool_arrrrr __from_arrow__fsF      zBooleanDtype.__from_arrow__N)__name__ __module__ __qualname____doc__r__annotations__propertyrr#r& classmethodr)r*r,r-r@rrrr r(s   rFr$tuple[np.ndarray, np.ndarray])copyrcCsBt|trD|durtd|j|j}}|r<|}|}||fSd}t|tjrp|jtj krp|rl|}nt|tjr|jj dvrt |}tj t |td}||t||<t|||j||kstd|}ntj|td}tj|dd}d}|d |vrtdtd t |}tj t |td}||t||<||vrt||t||tkstd|dur|durtj |jtd}np|dur|}n`t|tjr|jtj kr|dur||B}n|r$|}n tj|td}|dur$||B}|j|jkr:td ||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 inputZiufcbr/zNeed to pass bool-like valuesTskipna)Zfloatingintegerzmixed-integer-float)rr9npt.NDArray[np.bool_]z&values.shape and mask.shape must match)r5r' ValueError_data_maskrIrndarrayr%rr#r r;r6r$astypeallr4asarrayobjectrZ infer_dtyperfloatshaper.)valuesr=rIZ mask_valuesZ values_boolZ values_objectZinferred_dtypeZ integer_likerrr coerce_to_arraysh        rYc seZdZdZdZdZdZhdZhdZe dddd fd d Z d+ddd d dfdd Z e ddddZ e dddddddd ddddddZejejeejfZe dddd dd d!d"Zd#d$Zdd%d&d d'd(d)d*ZZS),r'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>1TRUEz1.0Truetrue>falsez0.0FALSE0Falsez np.ndarrayrMr)rXr=rcst||}t|_|Sr)super _simple_newr_dtype)r(rXr=result __class__rr rc-szBooleanArray._simple_newr$None)rXr=rIrcs>t|tjr|jtjks tdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrI) r5rrQr%rr4rrdrb__init__)rrXr=rIrfrr rj3s zBooleanArray.__init__rrcCs|jSr)rdrrrr r%>szBooleanArray.dtypeN)r%rI true_values false_valuesz list[str]z Dtype | Nonezlist[str] | None)stringsr%rIrkrlrc st|j|p g|j|pgddfdd }tj|td}t|}tt|||||<|j |||dS)Nr$rcs*|vr dS|vrdSt|ddS)NTFz cannot be cast to bool)rN)sZfalse_values_unionZtrue_values_unionrr map_stringOs z:BooleanArray._from_sequence_of_strings..map_stringr/)r%rI) _TRUE_VALUESunion _FALSE_VALUESrr.rUr listmapZ_from_sequence) r(rmr%rIrkrlrpZscalarsr=rror _from_sequence_of_stringsBs z&BooleanArray._from_sequence_of_stringsrirrH)r%rIrcCs|r|dksJt||dS)Nrri)rY)r(valuer%rIrrr _coerce_to_array^s zBooleanArray._coerce_to_arraycCs<|jdvsJt|}d}t|tr6|j|j}}nNt|rptj |dd}|j dkr^t dt |dd\}}nt|tj r|}|r|tjurt|stdt|jd |st|t|krtd |jd vrt|j||j|\}}n>|jd vrt|j||j|\}}nt|j||j|\}}|||S) N>Zrxorrand_xoror_and_ror_r$r/r0z(can only perform ops with 1-d structuresFriz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>r}r{>r|ry)rArZ is_scalarr5r'rOrPrrrTndimNotImplementedErrorrYritem libmissingZNAZis_boolr4rr6rNr Z kleene_orZ kleene_andZ kleene_xorZ_maybe_mask_result)rotheropZother_is_scalarr=rerrr _logical_methodfs4      zBooleanArray._logical_methodrJr!r )rrKrcKsv|j}|j}|dvrDtt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|SdS)N)ZcumminZcummaxrKr) IntegerArray) rOrPgetattrr rcZpandas.core.arraysrrRint _accumulate)rrrKkwargsr>r=rrrrr rs   zBooleanArray._accumulate)F)rArBrCrDZ_internal_fill_valueZ _truthy_valueZ _falsey_valuerqrsrGrcrjrFr%rvrrQnumbersNumberr$rZ_HANDLED_TYPESrxrr __classcell__rrrfr r's24 $r')NF)$ __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 rr2Zpandas._typingrrrrrrrYr'rrrr s$       pW