U c1+@sddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z mZmZddlmZmZddlmZddlmZdd lmZdd lmZmZerddlZdd l mZeGd d d eZddddddZGdddeZ dS)) annotationsN) TYPE_CHECKINGcast)libmissing)DtypeDtypeObjtype_t) is_list_likeis_numeric_dtype)register_extension_dtype)isna)ops)BaseMaskedArrayBaseMaskedDtype)nptc@seZdZdZdZeddddZedddd Zed dd d Ze d dddZ ddddZ eddddZ eddddZ dddddZdS) 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 booleantypereturncCstjSN)npbool_selfr>/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/arrays/boolean.pyrFszBooleanDtype.typestrcCsdS)NbrrrrrkindJszBooleanDtype.kindznp.dtypecCs tdS)Nbool)rdtyperrrr numpy_dtypeNszBooleanDtype.numpy_dtypeztype_t[BooleanArray]cCstS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clsrrrconstruct_array_typeRs z!BooleanDtype.construct_array_typecCsdS)Nrrrrrr__repr__]szBooleanDtype.__repr__r!cCsdSNTrrrrr _is_boolean`szBooleanDtype._is_booleancCsdSr(rrrrr _is_numericdszBooleanDtype._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+r/r3resultsZarrZbuflistdatamaskZbool_arrrrr__from_arrow__hsH      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 valuesT)Zskipna)Zfloatingintegerzmixed-integer-float)remptyznpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r1r$ ValueError_data_maskrDrndarrayr"rr r r6r5r!Zastypeallr0asarrayobjectrZ infer_dtyperfloatshaper+)valuesr:rDZ mask_valuesZ values_boolZ values_objectZinferred_dtypeZ integer_likerrrcoerce_to_arraysh        rQc seZdZdZdZdZdZdddddhZd d d d d hZd(dddddfdd Z e ddddZ e ddddddddddddddZ ejejeejfZe dd d!dd"d#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 FTTrueTRUEtrue1z1.0FalseFALSEfalse0z0.0z np.ndarrayr!None)rPr:rDrcs>t|tjr|jtjks tdt|_tj |||ddS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrD) r1rrJr"rr0r_dtypesuper__init__)rrPr:rD __class__rrr^'s zBooleanArray.__init__rrcCs|jSr)r\rrrrr"2szBooleanArray.dtypeN)r"rD true_values false_valuesz list[str]z Dtype | Nonezlist[str] | None)stringsr"rDrarbrcsP|j|p g|j|pgfddfdd|D}|j|||dS)Ncs6t|r |S|krdS|kr$dSt|ddS)NTFz cannot be cast to bool)r rG)s)false_values_uniontrue_values_unionrr map_stringCsz:BooleanArray._from_sequence_of_strings..map_stringcsg|] }|qSrr).0x)rgrr Msz:BooleanArray._from_sequence_of_strings..)r"rD) _TRUE_VALUESunion _FALSE_VALUESZ_from_sequence)r%rcr"rDrarbZscalarsr)rergrfr_from_sequence_of_strings6s  z&BooleanArray._from_sequence_of_stringsr[rrC)r"rDrcCs|r|dkstt||dS)Nrr[)AssertionErrorrQ)r%valuer"rDrrr_coerce_to_arrayRs 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>rand_ror_rxorxorand_or_r!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>rsrw>rvrr)r<rorZ is_scalarr1r$rHrIr rrLndimNotImplementedErrorrQritem libmissingZNAZis_boolr0rr5rGrZ kleene_orZ kleene_andZ kleene_xorZ_maybe_mask_result)rotheropZother_is_scalarr:resultrrr_logical_methodZs0      zBooleanArray._logical_method)F)r<r=r>r?Z_internal_fill_valueZ _truthy_valueZ _falsey_valuerkrmr^rAr"rBrnrrJnumbersNumberr!rZ_HANDLED_TYPESrqr __classcell__rrr_rr$s*6 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.arrays.maskedrrr/rrrQr$rrrrs&     jW