j6fddlmZddlZddlmZmZmZmZddlZ ddl m Z m Z ddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZerddlZdd lmZm Z m!Z!dd lm"Z"eedGddeZ# dddZ$edGddeZ%dS)) annotationsN) TYPE_CHECKINGClassVarSelfcast)libmissing) set_module) is_list_like)register_extension_dtype)isna)ops)masked_accumulations)BaseMaskedArrayBaseMaskedDtype)DtypeObjnpttype_t)ExtensionDtypepandasceZdZUdZdZded<dZeddZedd Z edd Z ddZ ddZ eddZ eddZddZdS) BooleanDtypeaP Extension dtype for boolean data. This is a pandas Extension dtype for boolean data with support for missing values. BooleanDtype is the dtype companion to :class:`.BooleanArray`, which implements Kleene logic (sometimes called three-value logic) for logical operations. See :ref:`boolean.kleene` for more. .. warning:: BooleanDtype is considered experimental. The implementation and parts of the API may change without warning. Attributes ---------- None Methods ------- None See Also -------- arrays.BooleanArray : Array of boolean (True/False) data with missing values. Int64Dtype : Extension dtype for int64 integer data. StringDtype : Extension dtype for string data. Examples -------- >>> pd.BooleanDtype() BooleanDtype >>> pd.array([True, False, None], dtype=pd.BooleanDtype()) [True, False, ] Length: 3, dtype: boolean >>> pd.array([True, False, None], dtype="boolean") [True, False, ] Length: 3, dtype: boolean booleanz ClassVar[str]nameFreturntypectjSN)npbool_selfs HC:\Projects\DSCons\.venv\Lib\site-packages\pandas\core\arrays\boolean.pyrzBooleanDtype.type_s xstrcdS)Nbr!s r#kindzBooleanDtype.kindcssr$np.dtypec*tjdS)Nbool)rdtyper!s r# numpy_dtypezBooleanDtype.numpy_dtypegsxr$type_t[BooleanArray]ctS)zq Return the array type associated with this dtype. Returns ------- type ) BooleanArrayr!s r#construct_array_typez!BooleanDtype.construct_array_typeks r$cdS)Nrr(r!s r#__repr__zBooleanDtype.__repr__us~r$r,cdSNTr(r!s r# _is_booleanzBooleanDtype._is_booleanxtr$cdSr6r(r!s r# _is_numericzBooleanDtype._is_numeric|r8r$array$pyarrow.Array | pyarrow.ChunkedArrayr1cddl}|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)pyarrowrr typesis_null TypeError isinstanceArraylenchunkslengthronesr,emptyr1buffers from_buffersr@to_numpy null_countzerosappendr;_concat_same_type) r"r;rBrIrJmaskdataresultsarrbuflistbool_arrs r#__from_arrow__zBooleanDtype.__from_arrow__sK  : ( (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<rr1)__name__ __module__ __qualname____doc__r__annotations___internal_fill_valuepropertyrr)r.r2r4r7r:rZr(r$r#rr*s))V$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 d g|Rvrt#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-floatrrLnpt.NDArray[np.bool_]z&values.shape and mask.shape must match)rFr1 ValueError_data_maskrbrndarrayr-r r)r rQrHr,astypeallrEasarrayobjectr infer_dtyperfloatshaper;)valuesrTrb mask_values values_bool values_objectinferred_dtype integer_likes r#coerce_to_arrayr|s0"&,''  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 )W!D|!D!D D D;<< <2D4G4GHH #f++T222,k\:AA$GG | l * * F |$++E22 +.55e<<=   + ;<< < | +x D111  D"* % %&$**@*@  "+%DD  99;;DxD)))  "+%D |tz!!ABBB 4<r$z pandas.arraysceZdZdZhdZhdZed)fd Z d*d+fd Ze d,dZ ed ddddd-dZ e j ejee jfZed dd.d Zd!Zd"d#d/d(ZxZS)0r1as 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 a 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 See Also -------- array : Create an array from data with the appropriate dtype. BooleanDtype : Extension dtype for boolean data. Series : One-dimensional ndarray with axis labels (including time series). DataFrame : Two-dimensional, size-mutable, potentially heterogeneous tabular data. Examples -------- Create a BooleanArray with :func:`pandas.array`: >>> pd.array([True, False, None], dtype="boolean") [True, False, ] Length: 3, dtype: boolean >1.01TRUETruetrue>0.00FALSEFalsefalserv np.ndarrayrTrjrrcrt||}t|_|Sr)super _simple_newr_dtype)clsrvrTresult __class__s r#rzBooleanArray._simple_newFs,$$VT22$  r$Frbr,Nonect|tjr|jtjkst dt |_t |||dS)NzIvalues should be boolean numpy array. Use the 'pd.array' function insteadrb) rFrrnr-r rErrr__init__)r"rvrTrbrs r#rzBooleanArray.__init__Lsp62:.. 6<283K3K2 #nn  D11111r$rc|jSr)rr!s r#r-zBooleanArray.dtypeWs {r$N)rb true_values false_values none_valuesstrings list[str]r-rrlist[str] | Nonerrcd |j|pg |j|pg gd  fd }tj|t }t |} tt||| || <| |||S)Nr bool | NonecL|vrdS|vrdS|vrdSt|d)NTFz cannot be cast to bool)rk)sfalse_values_unionrtrue_values_unions r# map_stringz:BooleanArray._from_sequence_of_strings..map_stringlsL%%%t(((uk!!t A!>!>!>???r$r>)r-rb)rr) _TRUE_VALUESunion _FALSE_VALUESrr;rrr listmap_from_sequence) rrr-rbrrrrscalarsrTrrs ` @@r#_from_sequence_of_stringsz&BooleanArray._from_sequence_of_strings[s ,22;3D"EE .44\5GRHH  K @ @ @ @ @ @ @ @(7&111G}}c*gten==>>!!'T!BBBr$rrrcc8|r|dksJt||S)Nrr)r|)rvaluer-rbs r#_coerce_to_arrayzBooleanArray._coerce_to_array}s0  &I%%%%u40000r$c|jdvsJtj|}d}t|tr|j|j}}nzt|r=tj |d}|j dkrtSt|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?Frz+'other' should be pandas.NA or a bool. Got z instead.zLengths must match>rr>rr)r[r is_scalarrFr1rlrmr rrqndimNotImplementedr|r item libmissingNAis_boolrErrHrkr kleene_or kleene_and kleene_xor_maybe_mask_result)r"otheropother_is_scalarrTrs r#_logical_methodzBooleanArray._logical_methods{MMMMM-.. e\ * * !+u{4EE %  !JuF333EzA~~%%)%e<<$*eTZNNLFDD>$*eTZNNLFD&&vt444r$Trfrr%rgrc  |j}|j}|dvr:tt|}|||fd|i|\}}|||Sddlm}||t|j |fd|i|S)N)cummincummaxrgr) IntegerArray) rlrmgetattrrrpandas.core.arraysrroint _accumulate)r"rrgkwargsrUrTrrs r#rzBooleanArray._accumulateszz ' ' '-t44BD$@@v@@@JD$##D$// / 7 7 7 7 7 7C<< C 0 0$77C#'- r$)rvrrTrjrr)F)rvrrTrrbr,rr)rr)rrr-rrbr,rrrrrrrr1)r-rrbr,rrc)rr%rgr,rr)r[r\r]r^rr classmethodrrrar-rrrnnumbersNumberr,r _HANDLED_TYPESrrr __classcell__)rs@r#r1r1sf88t877L;;;M[ BG 2 2 2 2 2 2 2X (,)-(,CCCCC[C>j'.$AN5:11111[1!5!5!5H,0r$r1)NF)rbr,rrc)& __future__rrtypingrrrrnumpyr pandas._libsrr rpandas.util._decoratorsr pandas.core.dtypes.commonr pandas.core.dtypes.dtypesr pandas.core.dtypes.missingr pandas.corerpandas.core.array_algosrpandas.core.arrays.maskedrrrBpandas._typingrrrrrr|r1r(r$r#rsO""""""/.....222222>>>>>>++++++888888  9NNN 988888 HC;C;C;C;C;?C;C;C;N%*RRRRRj Onnnnn?nnnnnr$