E g0ddlmZddlZddlmZmZmZddlZddl m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZmZerddlZdd lmZmZmZmZm Z eGd d eZ! dddZ"GddeZ#dS)) annotationsN) TYPE_CHECKINGClassVarcast)libmissing) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeDtypeObjSelfnpttype_tceZdZUdZdZded<eddZedd Zedd Z e dd Z ddZ eddZ eddZddZdS) 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]namereturntypectjSN)npbool_selfs O/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/core/arrays/boolean.pyrzBooleanDtype.typeDs xstrcdS)Nbrs r!kindzBooleanDtype.kindHssr"np.dtypec*tjdS)Nbool)rdtypers r! numpy_dtypezBooleanDtype.numpy_dtypeLsxr"type_t[BooleanArray]ctS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clss r!construct_array_typez!BooleanDtype.construct_array_typePs r"cdS)Nrr&rs r!__repr__zBooleanDtype.__repr__[s~r"r*cdSNTr&rs r! _is_booleanzBooleanDtype._is_boolean^tr"cdSr5r&rs r! _is_numericzBooleanDtype._is_numericbr7r"array$pyarrow.Array | pyarrow.ChunkedArrayr/cddl}|j|jkr7|j|jst d|jdt ||jr|g}t|}n|j }| }|j|jrFtj |t}tj|t}t||Sg}|D]}|} |j|jt|d| dg|jd }|jdkrU|j|jt|d| dg|jd }|}n(tjt|t}t||} || |sLttjgtjtjgtjSt|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_buffersr?to_numpy null_countzerosappendr:_concat_same_type) r r:rArHrImaskdataresultsarrbuflistbool_arrs r!__from_arrow__zBooleanDtype.__from_arrow__fsK  : ( (1F1Fuz1R1R (W5:WWWXX X eW] + + $WFZZFF\F\\^^F =  , , ,76...D8F$///Dd++ + % %CkkmmG'44#c((T71:$6sz5heh,, ~""+88Hc#hhwqz(:3:9(%(00uxC555#D$//H NN8 $ $ $ $ ;28,,,bhr.J.J.J  11':: :r"N)rr)rr#)rr()rr-rr*)r:r;rr/)__name__ __module__ __qualname____doc__r__annotations__propertyrr'r, classmethodr1r3r6r9rYr&r"r!rr(s,$D####XX   X [XX/;/;/;/;/;/;r"rFcopyr*rtuple[np.ndarray, np.ndarray]ct|trM|td|j|j}}|r(|}|}||fSd}t|t jr-|jt j kr|r|}nt|t jr|jj dvrt|}t j t|t}||t||<t j|||j||kst#d|}nt j|t&}t)j|d}d}|d |zvrt#dt-d t|}t j t|t}||t||<||vret j||t.||t.kst#d|#|!t j |jt}nu||}npt|t jr4|jt j kr|||z}n9|r|}n"t j|t}|||z}|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 inputiufcbr=zNeed to pass bool-like valuesTskipna)floatingintegerzmixed-integer-float)rrKnpt.NDArray[np.bool_]z&values.shape and mask.shape must match)rEr/ ValueError_data_maskrbrndarrayr+rr'r rPrGr*astypeallrDasarrayobjectr infer_dtyperfloatshaper:)valuesrSrb mask_values values_bool values_objectinferred_dtype integer_likes r!coerce_to_arrayr|s)"&,''  FGG G|V\  [[]]F99;;Dt|K&"*%%$=&,"(*B*B  #[[]]F FBJ ' '!=FL,=,H,H6ll hs6{{$777 $*K<$8$?$?$E$E [L!v  % , ,V\ : :fk\>R R   =;<< < 6888 tDDDE !5 !D D D;<< <2D4G4GHH #f++T222,k\:AA$GG | l * * F |$++E22 +.55e<<=   + ;<< < | +x D111  dBJ ' ' *DJ"(,B,B&k)'99;;D8D---D&k) |tz!!ABBB 4<r"ceZdZdZdZdZdZhdZhdZe d(fd Z d)d*fd Z e d+dZ e dddddd,dZejejeejfZe ddd-d Zd!Zdd"d.d'ZxZS)/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>1.01TRUETruetrue>0.00FALSEFalsefalserv np.ndarrayrSrjrrcrt||}t|_|Sr)super _simple_newr_dtype)r0rvrSresult __class__s r!rzBooleanArray._simple_new-s,$$VT22$  r"rbr*Nonect|tjr|jtjkst dt |_t |||dS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrb) rErrnr+rrDrrr__init__)r rvrSrbrs r!rzBooleanArray.__init__3sp62:.. 6<283K3K2 #nn  D11111r"rc|jSr)rrs r!r+zBooleanArray.dtype>s {r"N)r+rb true_values false_valuesstrings list[str]r+ Dtype | Nonerlist[str] | NonercX |j|pg |j|pg d fd }tj|t }t |}tt|||||<| |||S)Nrr*c@|vrdS|vrdSt|d)NTFz cannot be cast to bool)rk)sfalse_values_uniontrue_values_unions r! map_stringz:BooleanArray._from_sequence_of_strings..map_stringOs<%%%t(((u A!>!>!>???r"r=)r+rbrZ) _TRUE_VALUESunion _FALSE_VALUESrr:rrr listmap_from_sequence) r0rr+rbrrrscalarsrSrrs @@r!_from_sequence_of_stringsz&BooleanArray._from_sequence_of_stringsBs ,22;3D"EE .44\5GRHH @ @ @ @ @ @ @(7&111G}}c*gten==>>!!'T!BBBr"rrrcc8|r|dksJt||S)Nrr)r|)r0valuer+rbs r!_coerce_to_arrayzBooleanArray._coerce_to_array^s0  &I%%%%u40000r"c|jdvsJtj|}d}t|tr|j|j}}nt|rEtj |d}|j dkrtdt|d\}}n.t|tj r|}|rG|tjur9tj|s%t%dt'|jd |s/t)|t)|krt+d |jd vr%t-j|j||j|\}}nR|jd vr%t-j|j||j|\}}n$t-j|j||j|\}}|||S) N>or_xorand_ror_rxorrand_r*r=r>z(can only perform ops with 1-d structuresFrz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)r[r is_scalarrEr/rlrmr rrqndimNotImplementedErrorr|ritem libmissingNAis_boolrDrrGrkr kleene_or kleene_and kleene_xor_maybe_mask_result)r otheropother_is_scalarrSrs r!_logical_methodzBooleanArray._logical_methodfs{MMMMM-.. e\ * * !+u{4EE %  !JuF333EzA~~)*TUUU)%e<<$*eTZNNLFDD>$*eTZNNLFD&&vt444r"rfrr#rgrc  |j}|j}|dvr:tt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|S)N)cummincummaxrgr) IntegerArray) rlrmgetattrr rpandas.core.arraysrroint _accumulate)r rrgkwargsrTrSrrs r!rzBooleanArray._accumulateszz ' ' '-t44BD$@@v@@@JD$##D$// / 7 7 7 7 7 7C<< C 0 0$77C#'- r")rvrrSrjrr)F)rvrrSrrbr*rr)rr) rrr+rrbr*rrrrrr/)r+rrbr*rrc)rr#rgr*rr)r[r\r]r^_internal_fill_value _truthy_value _falsey_valuerrrarrr`r+rrrnnumbersNumberr*r_HANDLED_TYPESrrr __classcell__)rs@r!r/r/sx11h!MM777L;;;M[ BG 2 2 2 2 2 2 2X #(,)-CCCCC[C2j'.$AN5:11111[1!5!5!5H,0r"r/)NF)rbr*rrc)$ __future__rrtypingrrrnumpyr pandas._libsrrrpandas.core.dtypes.commonr pandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corer pandas.core.array_algosr pandas.core.arrays.maskedrrrApandas._typingrrrrrrr|r/r&r"r!rs""""""  322222>>>>>>++++++888888  NNNl;l;l;l;l;?l;l;l;`%*TTTTTnhhhhh?hhhhhr"