U MZff.@sddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z mZmZddlmZmZddlmZddlmZdd lmZdd lmZdd lmZmZerddlZdd l mZeGd ddeZ ddddddZ!GdddeZ"dS)) annotationsN) TYPE_CHECKINGcast)libmissing)DtypeDtypeObjtype_t) is_list_likeis_numeric_dtype)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)nptc@seZdZdZdZeddddZedddd Zed dd d Ze d dddZ ddddZ eddddZ eddddZ dddddZdS) 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 booleantypereturncCstjSN)npbool_selfr>/tmp/pip-unpacked-wheel-nbcvw55c/pandas/core/arrays/boolean.pyrEszBooleanDtype.typestrcCsdS)NbrrrrrkindIszBooleanDtype.kindznp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeMszBooleanDtype.numpy_dtypeztype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrrconstruct_array_typeQs z!BooleanDtype.construct_array_typecCsdS)Nrrrrrr__repr__\szBooleanDtype.__repr__r"cCsdSNTrrrrr _is_boolean_szBooleanDtype._is_booleancCsdSr)rrrrr _is_numericcszBooleanDtype._is_numericz$pyarrow.Array | pyarrow.ChunkedArrayr%)arrayrc Csddl}|j|kr(td|jdt||jr<|g}n|j}g}|D]}|}|jj |jt |d|dg|j dj dd}|j dkr|jj |jt |d|dg|j dj dd}|}ntjt |td }t||} || qJ|sttjgtjd tjgtjd St|SdS) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z instead)offsetF)Zzero_copy_onlyr#)pyarrowrr TypeError isinstanceArraychunksbuffersr%Z from_bufferslenr.Zto_numpyZ null_countrzerosr"appendr,Z_concat_same_type) rr,r0r4resultsZarrZbuflistdatamaskZbool_arrrrr__from_arrow__gsH      zBooleanDtype.__from_arrow__N)__name__ __module__ __qualname____doc__namepropertyrr!r$ classmethodr'r(r*r+r<rrrrr)s  rFr"tuple[np.ndarray, np.ndarray])copyrcCs@t|trD|dk rtd|j|j}}|r<|}|}||fSd}t|tjrp|jtj krp|rl|}nt|tjrt |jrt |}tj t |td}||t||<t|||j||kstd|}ntj|td}tj|dd}d}|d|krtdtd t |}tj t |td}||t||<||krt||t||tkstd|dkr|dkrtj |jtd}np|dkr|}n`t|tjr|jtj kr|dk r||B}n|r"|}n tj|td}|dk r"||B}|j|jkr8td ||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 inputr/zNeed to pass bool-like valuesTskipna)Zfloatingintegerzmixed-integer-float)remptyznpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r2r% ValueError_data_maskrErndarrayr#rr r r7r6r"astypeallr1asarrayobjectrZ infer_dtyperfloatshaper,)valuesr;rEZ mask_valuesZ values_boolZ values_objectZinferred_dtypeZ integer_likerrrcoerce_to_arraysh        rUc seZdZdZdZdZdZdddddhZd d d d d hZd.dddddfdd Z e ddddZ e ddddddddddddddZ ejejeejfZe dd d!dd"d#d$d%Zd&d'Zdd(d)dd*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 FTTrueTRUEtrue1z1.0FalseFALSEfalse0z0.0z np.ndarrayr"None)rTr;rErcs>t|tjr|jtjks tdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrE) r2rrMr#rr1r_dtypesuper__init__)rrTr;rE __class__rrrb&s zBooleanArray.__init__rrcCs|jSr)r`rrrrr#1szBooleanArray.dtypeN)r#rE true_values false_valuesz list[str]z Dtype | Nonezlist[str] | None)stringsr#rErerfrc st|j|p g|j|pgddfdd }tj|td}t|}tt|||||<|j |||dS)Nr"rcs*|kr dS|krdSt|ddS)NTFz cannot be cast to bool)rJ)sZfalse_values_unionZtrue_values_unionrr map_stringBs z:BooleanArray._from_sequence_of_strings..map_stringr/)r#rE) _TRUE_VALUESunion _FALSE_VALUESrr,rQr listmapZ_from_sequence) r&rgr#rErerfrjZscalarsr;rrir_from_sequence_of_strings5s z&BooleanArray._from_sequence_of_stringsr_rrD)r#rErcCs|r|dkstt||dS)Nrr_)AssertionErrorrU)r&valuer#rErrr_coerce_to_arrayQs zBooleanArray._coerce_to_arraycCs<|jdkstt|}d}t|tr6|j|j}}nNt|rpt j |dd}|j dkr^t dt |dd\}}nt|t jr|}|r|tjk rt|stdt|jd |st|t|krtd |jd krt|j||j|\}}n>|jd krt|j||j|\}}nt|j||j|\}}|||S) N>and_rand_or_rxorror_xorr"r/r-z(can only perform ops with 1-d structuresFr_z+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rxrv>rtru)r=rqrZ is_scalarr2r%rKrLr rrPndimNotImplementedErrorrUritem libmissingZNAZis_boolr1rr6rJrZ kleene_orZ kleene_andZ kleene_xorZ_maybe_mask_result)rotheropZother_is_scalarr;resultrrr_logical_methodYs0      zBooleanArray._logical_methodrFrr)rArGrcKs||j}|j}|dkrJtt|}|||fd|i|\}}t|||ddSddlm}||t|j |fd|i|SdS)N)ZcumminZcummaxrGFr_r) IntegerArray) rKrLgetattrrrZpandas.core.arraysrrNint _accumulate)rrArGkwargsr:r;rrrrrr|s  zBooleanArray._accumulate)F)r=r>r?r@Z_internal_fill_valueZ _truthy_valueZ _falsey_valuerkrmrbrBr#rCrprrMnumbersNumberr"rZ_HANDLED_TYPESrsrr __classcell__rrrcrr%s.4 $r%)NF)# __future__rrtypingrrZnumpyrZ pandas._libsrrr}Zpandas._typingrrr Zpandas.core.dtypes.commonr r Zpandas.core.dtypes.dtypesr Zpandas.core.dtypes.missingr Z pandas.corerZpandas.core.array_algosrZpandas.core.arrays.maskedrrr0rrrUr%rrrrs(      hW