`Vg1+ddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z mZmZddlmZmZddlmZddlmZdd lmZdd lmZmZer ddlZdd l mZeGd d eZ d ddZGddeZ y)) annotationsN) TYPE_CHECKINGcast)libmissing)DtypeDtypeObjtype_t) is_list_likeis_numeric_dtype)register_extension_dtype)isna)ops)BaseMaskedArrayBaseMaskedDtype)nptceZdZdZdZed dZed dZeddZe ddZ d dZ eddZ edd Z dd Zy ) 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 booleanc"tjSN)npbool_selfs [/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/pandas/core/arrays/boolean.pytypezBooleanDtype.typeFs xxcy)Nbrs rkindzBooleanDtype.kindJsrc,tjdS)Nbool)rdtypers r numpy_dtypezBooleanDtype.numpy_dtypeNsxxrctS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clss rconstruct_array_typez!BooleanDtype.construct_array_typeRs rcy)Nrr!rs r__repr__zBooleanDtype.__repr__]srcyNTr!rs r _is_booleanzBooleanDtype._is_boolean`rcyr.r!rs r _is_numericzBooleanDtype._is_numericdr0rcddl}|j|jk7rtd|jdt ||j r|g}n |j }g}|D] }|j}|jj|jt|d|dg|jjd}|jdk7rU|jj|jt|d|dg|jjd}|}n$tjt|t }t||} |j#|  |sRttj$gtj tj$gtj Stj'|S) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z instead)offsetF)zero_copy_onlyr%)pyarrowrr TypeError isinstanceArraychunksbuffersr( from_bufferslenr5to_numpy null_countrzerosr$appendarray_concat_same_type) rrDr8r<resultsarrbuflistdatamaskbool_arrs r__from_arrow__zBooleanDtype.__from_arrow__hs  :: (B5::,hWX X eW]] +WF\\F %CkkmG''44#c(T71:$6szz5heh, ~~"++88HHc#hwqz(:3::9(%(0uxxC5#D$/H NN8 $ % 288,bhhr.J  11': :rN)returnr)rMstr)rMznp.dtype)rMztype_t[BooleanArray])rMr$)rDz$pyarrow.Array | pyarrow.ChunkedArrayrMr()__name__ __module__ __qualname____doc__namepropertyrr"r& classmethodr*r,r/r2rLr!rrrr(s0 D  ';9'; ';rrcPt|trK| td|j|j}}|r |j }|j }||fSd}t|t jr2|jt jk(r|r|j }nt|t jrt|jrt|}t jt|t}||jt||<t j ||j|j||k(s t#d|}nt j$|t&}t)j*|d}d}|d|zvr t#dt-d t|}t jt|t}||jt||<||vrQt j ||jt.||jt.k(s t#d|(|&t j|j0t}ny||}ntt|t jr8|jt jk(r|||z}n5|r3|j }n"t j2|t}|||z}|j0|j0k7r 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 inputr7zNeed to pass bool-like valuesT)skipna)floatingintegerzmixed-integer-float)remptyznpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r:r( ValueError_data_maskcopyrndarrayr%rr rrBr?r$astypeallr9asarrayobjectr infer_dtyperfloatshaperD)valuesrJr^ mask_values values_bool values_objectinferred_dtype integer_likes rcoerce_to_arrayrms"&,'  FG G||V\\ [[]F99;Dt|K&"**%&,,"((*B [[]F FBJJ ',R R ;< < 68 tDE !5 !D D;< <2D4GH #f+T2,k\:AA$G | l * FF |$++E2 +.55e<=  ;< < | +xx D1  dBJJ 'DJJ"((,B&k)99;D88D-D&k) ||tzz!ABB 4<rceZdZdZdZdZdZhdZhdZ d dfd Z e ddZ e ddddd  dd Z ejej"eej&fZe dd  dd Zd ZxZS)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 FT>1.01TRUETruetrue>0.00FALSEFalsefalsect|tjr|jtjk(s t dt |_t|%|||y)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadr^) r:rr_r%rr9r_dtypesuper__init__)rrgrJr^ __class__s rr}zBooleanArray.__init__'sS62::.6<<2883K2 #n  D1rc|jSr)r{rs rr%zBooleanArray.dtype2s {{rN)r%r^ true_values false_valuesc |jj|xsg |jj|xsg fd}|Dcgc] }|| }}|j|||Scc}w)NcNt|r|S|vry|vryt|d)NTFz cannot be cast to bool)rr[)sfalse_values_uniontrue_values_unions r map_stringz:BooleanArray._from_sequence_of_strings..map_stringCs:Aw''(( A3&=!>??r)r%r^) _TRUE_VALUESunion _FALSE_VALUES_from_sequence) r)stringsr%r^rrrxscalarsrrs @@r_from_sequence_of_stringsz&BooleanArray._from_sequence_of_strings6sy ,,22;3D"E ..44\5GRH @+22Q:a=22!!'T!BB3s A0rzc.|r|dk(sJt||S)Nrrz)rm)r)valuer%r^s r_coerce_to_arrayzBooleanArray._coerce_to_arrayRs" I% %%u400rc|jdvsJtj|}d}t|tr|j |j }}nwt|rBtj|d}|jdkDr tdt|d\}}n*t|tjr|j}|rI|tj ur7tj"|s"t%dt'|jd |s"t)|t)|k7r t+d |jd vr0t-j.|j ||j |\}}nm|jd vr0t-j0|j ||j |\}}n/t-j2|j ||j |\}}|j5||S) N>or_xorand_ror_rxorrand_r$r7r4z(can only perform ops with 1-d structuresFrzz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rOr is_scalarr:r(r\r]r rrbndimNotImplementedErrorrmritem libmissingNAis_boolr9rr?r[r kleene_or kleene_and kleene_xor_maybe_mask_result)rotheropother_is_scalarrJresults r_logical_methodzBooleanArray._logical_methodZs{{MMMM--. e\ *++u{{4E % JJuF3EzzA~)*TUU)%e>$**eTZZNLFD>>$**eTZZNLFD&&vt44r)F)rg np.ndarrayrJrr^r$rMNone)rMr) rz list[str]r%z Dtype | Noner^r$rlist[str] | NonerrrMr()r%r r^r$rMtuple[np.ndarray, np.ndarray])rOrPrQrR_internal_fill_value _truthy_value _falsey_valuerrr}rTr%rUrrr_numbersNumberr$r_HANDLED_TYPESrr __classcell__)r~s@rr(r(s3l!MM7L;MBG 2  2(2 2:> 2  2 #(,)-CC C  C & C'C CC2jj'..$AN5:1&1.21 &11"5rr()NF)r^r$rMr)! __future__rrtypingrrnumpyr pandas._libsrrrpandas._typingrr r pandas.core.dtypes.commonr r pandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corerpandas.core.arrays.maskedrrr8rrrmr(r!rrrs"  ?+ "f;?f;f;T%*T!T"TnS5?S5r