eh0ddlmZddlZddlmZmZmZddlZddl m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZmZerddlZdd lmZmZmZmZm Z eGd d eZ! d ddZ"GddeZ#y)) annotationsN) TYPE_CHECKINGClassVarcast)libmissing) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeDtypeObjSelfnpttype_tceZdZUdZdZded<eddZeddZeddZ e ddZ dd Z edd Z edd Z dd Zy ) 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]namec"tjSN)npbool_selfs X/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/arrays/boolean.pytypezBooleanDtype.typeDs xxcy)Nbrs rkindzBooleanDtype.kindHsr!c,tjdS)Nbool)rdtypers r numpy_dtypezBooleanDtype.numpy_dtypeLsxxr!ctS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArray)clss rconstruct_array_typez!BooleanDtype.construct_array_typePs r!cy)Nrr$rs r__repr__zBooleanDtype.__repr__[sr!cyNTr$rs r _is_booleanzBooleanDtype._is_boolean^r!cyr1r$rs r _is_numericzBooleanDtype._is_numericbr3r!cddl}|j|jk7r>|jj |jst d|jdt ||jr|g}t|}n|j}|j}|jj |jrBtj|t}tj|t}t||Sg}|D] }|j!} |jj#|jt|d| dg|j$j'd }|j(dk7rU|jj#|jt|d| dg|j$j'd }|}n$tj*t|t}t||} |j-|  |sRttj.gtjtj.gtjStj1|S) zI Construct BooleanArray from pyarrow Array/ChunkedArray. rNz$Expected array of boolean type, got z insteadr()offsetF)zero_copy_only)pyarrowr rtypesis_null TypeError isinstanceArraylenchunkslengthronesr'emptyr+buffers from_buffersr9to_numpy null_countzerosappendarray_concat_same_type) rrLr;rBrCmaskdataresultsarrbuflistbool_arrs r__from_arrow__zBooleanDtype.__from_arrow__fs  :: (1F1Fuzz1RB5::,hWX X eW]] +WFZF\\F\\^F ==  ,776.D88F$/Dd+ + %CkkmG''44#c(T71:$6szz5heh, ~~"++88HHc#hwqz(:3::9(%(0uxxC5#D$/H NN8 $ % 288,bhhr.J  11': :r!N)returnr )rUstr)rUznp.dtype)rUztype_t[BooleanArray]rUr')rLz$pyarrow.Array | pyarrow.ChunkedArrayrUr+)__name__ __module__ __qualname____doc__r__annotations__propertyr r%r) classmethodr-r/r2r5rTr$r!rrr(s,$D-#  /;9/; /;r!rcVt|trK| td|j|j}}|r |j }|j }||fSd}t|t jr2|jt jk(r|r|j }nt|t jr|jjdvrt|}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 inputiufcbr7zNeed to pass bool-like valuesTskipna)floatingintegerzmixed-integer-float)rrEnpt.NDArray[np.bool_]z&values.shape and mask.shape must match)r?r+ ValueError_data_maskcopyrndarrayr(rr%r rJrAr'astypeallr>asarrayobjectr infer_dtyperfloatshaperL)valuesrNri mask_values values_bool values_objectinferred_dtype integer_likes rcoerce_to_arrayrxs"&,'  FG G||V\\ [[]F99;Dt|K&"**%&,,"((*B [[]F FBJJ 'FLL,=,=,H6l hhs6{$7 $*K<$8$?$?$E [L!vv  % , ,V\\ :fk\>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<r!c6eZdZdZdZdZdZhdZhdZe dfd Z d dfd Z e ddZ e d dd d d  dd Zej ej$eej(fZe dd  dd ZdZdd ddZxZS)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.00FALSEFalsefalsecFt|||}t|_|Sr)super _simple_newr_dtype)r,rrrNresult __class__s rrzBooleanArray._simple_new-s"$VT2$  r!ct|tjr|jtjk(s t dt |_t|%|||y)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadri) r?rrjr(rr>rrr__init__)rrrrNrirs rrzBooleanArray.__init__3sS62::.6<<2883K2 #n  D1r!c|jSr)rrs rr(zBooleanArray.dtype>s {{r!N)r(ri true_values false_valuesc> |jj|xsg |jj|xsg d fd }tj|t }t |}tt|||||<|j|||S)Nc4|vry|vryt|d)NTFz cannot be cast to bool)rf)sfalse_values_uniontrue_values_unions r map_stringz:BooleanArray._from_sequence_of_strings..map_stringOs.%%(( A3&=!>??r!r7)r(rirW) _TRUE_VALUESunion _FALSE_VALUESrrLrnr listmap_from_sequence) r,stringsr(rirrrscalarsrNrrs @@r_from_sequence_of_stringsz&BooleanArray._from_sequence_of_stringsBs ,,22;3D"E ..44\5GRH @((7&1G}c*gten=>!!'T!BBr!rc.|r|dk(sJt||S)Nrr)rx)r,valuer(ris r_coerce_to_arrayzBooleanArray._coerce_to_array^s" I% %%u400r!c|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'r7r8z(can only perform ops with 1-d structuresFrz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)rXr is_scalarr?r+rgrhr rrmndimNotImplementedErrorrxritem libmissingNAis_boolr>r rArfr kleene_or kleene_and kleene_xor_maybe_mask_result)rotheropother_is_scalarrNrs r_logical_methodzBooleanArray._logical_methodfs{{MMMM--. e\ *++u{{4E % JJuF3EzzA~)*TUU)%e>$**eTZZNLFD>>$**eTZZNLFD&&vt44r!rac |j}|j}|dvr1tt|}|||fd|i|\}}|j ||Sddlm}||jt|j|fd|i|S)N)cummincummaxrbr) IntegerArray) rgrhgetattrr rpandas.core.arraysrrkint _accumulate)rrrbkwargsrOrNrrs rrzBooleanArray._accumulateszzzz ' '-t4BD$@v@@JD$##D$/ / 7C< C 0$7CC#'- r!)rr np.ndarrayrNrerUr)F)rrrrNrrir'rUNone)rUr) rz list[str]r(z Dtype | Nonerir'rlist[str] | NonerrrUr+)r(rrir'rUtuple[np.ndarray, np.ndarray])rrVrbr'rUr)rXrYrZr[_internal_fill_value _truthy_value _falsey_valuerrr^rrr]r(rrrjnumbersNumberr'r_HANDLED_TYPESrrr __classcell__)rs@rr+r+sN1h!MM7L;M BG 2  2(2 2:> 2  2 #(,)-CC C  C & C'C CC2jj'..$AN5:1&1.21 &11!5H,0$( r!r+)NF)rir'rUr)$ __future__rrtypingrrrnumpyr pandas._libsrrrpandas.core.dtypes.commonr pandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corer pandas.core.array_algosr pandas.core.arrays.maskedrrr;pandas._typingrrrrrrrxr+r$r!rrs"  3>+8 l;?l;l;`%*T!T"Tnh?hr!