hdZddlmZddlZddlmZmZmZddlZ ddl m Z m Z ddl mZmZddlmZmZmZmZddlmZmZmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#m$Z$ddl%m&cm'Z(dd l)m*Z*dd l+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;ddlm?Z?m@Z@mAZAmBZBddlCmDZDmEZEmFZFmGZGmHZHmIZIddlJmKZKerddlLmMZMddlNmOZOmPZPmQZQmRZRmSZSmTZTGdde?ZUGddeUZVGddeUe@ZWGddZXd!d ZYy)"zA Experimental manager based on storing a collection of 1D arrays ) annotationsN) TYPE_CHECKINGCallableLiteral)NaTlib) astype_arrayastype_array_safe)ensure_dtype_can_hold_nafind_common_typeinfer_dtype_from_scalarnp_find_common_type)ensure_platform_intis_datetime64_ns_dtype is_integeris_numeric_dtypeis_object_dtypeis_timedelta64_ns_dtype)ExtensionDtype) ABCDataFrame ABCSeries) array_equalsisnana_value_for_dtype)quantile_compat)take_1d) DatetimeArrayExtensionArrayNumpyExtensionArrayTimedeltaArray)ensure_wrapped_if_datetimelike extract_arraysanitize_array)maybe_convert_indicesvalidate_indices)Index ensure_index)get_values_for_csv) DataManagerSingleDataManagerensure_np_dtypeinterleaved_dtype)BlockPlacementensure_block_shapeexternal_valuesextract_pandas_arraymaybe_coerce_values new_block) make_na_array)Hashable) ArrayLikeAxisIntDtypeObjQuantileInterpolationSelfnptceZdZUdZddgZded<ded< d( d)dZd*d+dZed,d Z ed-d Z ed.d Z e d/d Z d0d Zd1dZd2dZdZd3dZd4dZ d* d5dZd*d+dZd(d6dZd7dZd8d9dZd:dZddd d;dZedd?d Zd>d?d!Z d(d@d"Z! dA dBd#Z" dC dDd$Z# dE dFd%Z$dGdHd&Z%daxesverify_integritys HD:\jyotish\venv\Lib\site-packages\pandas/core/internals/array_manager.py__init__zBaseArrayManager.__init__~s "!Ncb||jddtgg}g}t|||S)zFReturn an empty ArrayManager with the items axis of len 0 (no columns)N)rFr&type)rErFr>s rH make_emptyzBaseArrayManager.make_emptys7 <IIabM59-D46tDz&$''rJc |jdS)Nr=rEs rHitemszBaseArrayManager.itemsszz"~rJc>|jd|jdgS)z5Axes is BlockManager-compatible order (columns, rows)rLrrQrRs rHrFzBaseArrayManager.axess  1 tzz!}--rJc:td|jDS)Nc32K|]}t|ywrB)len).0axs rH z0BaseArrayManager.shape_proper..s2zSWzs)tupler=rRs rH shape_properzBaseArrayManager.shape_propers2tzz222rJc|dk(rd}|Sd}|SNrrLaxiss rH_normalize_axisz BaseArrayManager._normalize_axiss AIq $% rJch|j|||j|}||j|<yrB)_validate_set_axisrbr=)rEra new_labelss rHset_axiszBaseArrayManager.set_axiss0 j1##D)% 4rJc~tj|jDcgc]}|jc}dScc}w)Nobjectdtype)nparrayr>rjrEarrs rH get_dtypeszBaseArrayManager.get_dtypess-xxdkk:ksk:(KK:s:cy)z< Only implemented on the BlockManager level Nr_)rEmgrs rHadd_referenceszBaseArrayManager.add_referencess rJc2|j|jfSrBr>r=rRs rH __getstate__zBaseArrayManager.__getstate__s{{DJJ&&rJc,|d|_|d|_yr^rt)rEstates rH __setstate__zBaseArrayManager.__setstate__sAh 1X rJc t|j}|d|jdz }|jdk(r|d|jdz }|dt |j dz }|j D]}|d|j z }|S)Nz Index: rz Columns: rL z arrays:)rM__name__r=ndimrWr>rj)rEoutputrns rH__repr__zBaseArrayManager.__repr__sd$$Idjjm_-- 99>  DJJqM?3 3FBs4;;'(11;;C 399+& &F rJc Vd|vsJ|xsg}g}|Dcic]}||| }}|dk(r|jd}t|jD]\}}|ry|jD]f\}} t | t t frC| jdk(r| j|||<>| jdd|fj||<_| |||<ht|r ||fi|} nt||di|} |j| |j} t||| Scc}w)ab Iterate over the arrays, collect and create a new ArrayManager. Parameters ---------- f : str or callable Name of the Array method to apply. align_keys: List[str] or None, default None **kwargs Keywords to pass to `f` Returns ------- ArrayManager filterapplyfuncrLNr_)pop enumerater>rS isinstancerrr}iloc_valuescallablegetattrappendr=rM) rEf align_keyskwargs result_arraysk aligned_argsirnobjappliednew_axess rHrzBaseArrayManager.applys:*v%%%%2 )+ /99j6!9 j 9 < 6"A ,FAs*002FAs!# <'@A88q=(+ F1I(+A(>(>F1I%(Fq 3{C*6*)'#q/3F3   )'-*::tDz-227:s D&c >d}|dk(rd}|r"d|vr|jdk(r|ddk(rdnd|d<|xsg}|Dcic]}||| }}g}t|jD]\}} |r|jD]\}} t | t t fr| jdk(rV|jdk(r*| jt||dzj||<d| jddj||<| jdd|gfj||<| jdk(s| |g||<t | jtjr/t | tjstj| } t| } |jdk(r1t| d} t!tddd} t#| | d} n-t!tdt%|d} t#| | d} t'| |d i|} t | t(r| d} | j*} |jdk(r(| jdk(rt%| dk(sJ| dddf} |j-| t/|||j0Scc}w) NT interpolateFrarzrrL) placementr}r_)r}rr>rSrrrrslicerrjrkndarrayasarrayr1r.r-r2rWrlistvaluesrrMr=)rErrr swap_axisrrrrrnrbpblockrs rHapply_with_blockz!BaseArrayManager.apply_with_blocks_ I 6)dii1n"(.A"5Q1F6N%2 .89j6!9 j 9  ,FAs*002FAs!# <'@A88q=#yyA~,/HHU1a!e_,E,M,Mq ,/HHQK,?,?q (+QC(8(@(@F1I88q=(+QCF1I3 #))RXX.z#rzz7Rjjo%c*CyyA~(a0#E!QN3!#!<#E!SY$:;!#!<'geQ'1&1G'4(!!*..CyyA~#((a-3x1}$}!Q$i   %U-XtDz-44a:s Jc*|jd||S)Nsetitemindexervalue)rrErrwarns rHrzBaseArrayManager.setitem8s$$Yu$MMrJcf|jdk(sJ|jtj|S)Nrzn)r}ralgosdiffrErs rHrzBaseArrayManager.diff;s)yyA~~zz%**z**rJc<|d}|jt|||S)NT)rjcopyerrors)rr )rErjrrs rHastypezBaseArrayManager.astype?s$ <Dzz+5tFzSSrJc8dfd}|j|S)NTct|jrDtj|}t j |d}||urr|j S|Sr|j S|S)NT)convert_non_numeric)rrjrkrrmaybe_convert_objectsr)rnresultrs rH_convertz*BaseArrayManager.convert.._convertIscsyy)jjo22(,S=T88:% %)sxxz2s2rJ)r)rErrs ` rHconvertzBaseArrayManager.convertEs# <D 3zz(##rJnan)na_repquotingc8|jt|||||S)N)rr float_format date_formatdecimal)rr()rErrrrrs rHr(z#BaseArrayManager.get_values_for_csvZs,zz %#   rJcy)z>Whether any of the blocks in this manager are extension blocksFr_rRs rHany_extension_typesz$BaseArrayManager.any_extension_typesfsrJcy)z8return a boolean if we are a single block and are a viewFr_rRs rHis_viewzBaseArrayManager.is_viewksrJc2t|jdk(SNrL)rWr>rRs rHis_single_blockz BaseArrayManager.is_single_blockqs4;;1$$rJcdt|jDcgc]\}}||s|}}}|Dcgc]}|j|}}tj|d}|jdj |}|jd|g}t |||dScc}}wcc}w)NintprirLrFrG)rr>rkrlr=takerM) rE predicaterrnindicesr>takernew_colsrs rH_get_data_subsetz!BaseArrayManager._get_data_subsetus#,T[[#9L#9CYs^1#9L*12'Q$++a.'2/::a=%%e,JJqM8,tDz&(UCCM2sB'B'B-c&|jdS)z Select columns that are bool-dtype and object-dtype columns that are all-bool. Parameters ---------- copy : bool, default False Whether to copy the blocks cN|jtjtk(SrB)rjrkbool)xs rHz0BaseArrayManager.get_bool_data..sqww"((4./HrJrrErs rH get_bool_datazBaseArrayManager.get_bool_datas$$%HIIrJc&|jdS)z Select columns that have a numeric dtype. Parameters ---------- copy : bool, default False Whether to copy the blocks c^t|jxst|jddS)N _is_numericF)rrjr)rns rHrz3BaseArrayManager.get_numeric_data..s((38syy-78rJrrs rHget_numeric_dataz!BaseArrayManager.get_numeric_datas$$ 8  rJc>dr%fd}|jDcgc] }|| }}nt|j}r(|jDcgc]}|j}}nt|j}t |||dScc}wcc}w)a> Make deep or shallow copy of ArrayManager Parameters ---------- deep : bool or string, default True If False, return shallow copy (do not copy data) If 'all', copy data and a deep copy of the index Returns ------- BlockManager TcRdk(r|jdS|jS)NallTdeep)rview)rYrs rH copy_funcz(BaseArrayManager.copy..copy_funcs%-1U]rwwDw)I IrJFr)r=rr>rrM)rErrrYrrn new_arrayss ` rHrzBaseArrayManager.copys <D  J15 ; " " H;DJJ'H 04 < #((* J<dkk*JtDz*hGG< =s BBc R|j|}|j|||||||SrB)rb_reindex_indexer) rEnew_axisrra fill_value allow_dupsr only_slice use_na_proxys rHreindex_indexerz BaseArrayManager.reindex_indexers<##D)$$         rJc |d}|R||j|ur|s|S|j|}t|j|_||j|<|S|s|j|j|||jk\r t d|dk(rUg} |D]M} | dk(r|j ||} n!|j| } |r| j} | j| Onit|t|jdt|}|dk(} | j} |jD cgc]} t| || || } } t|j}|||<t|| |d Scc} w) a! Parameters ---------- new_axis : Index indexer : ndarray[intp] or None axis : int fill_value : object, default None allow_dups : bool, default False copy : bool, default True pandas-indexer with -1's only. Tr#Requested axis not found in managerrLrP)rrr allow_fillrmaskFr)r=rr_validate_can_reindexr} IndexError_make_na_arrayr>rr%rWranyrrM)rErrrarrrrrrrrnr needs_maskingrs rHrz!BaseArrayManager._reindex_indexers. <D ?4::d++D YYDY)F +FL!)FLL M JJt  2 27 ; 499 BC C 19J7--#-L.C++a.C!hhj!!#& Wc$**Q-&8 9)'2Gb=D HHJM ;; 'C,) '   #!tDz*hGG s=Fct|tjsJt||jtj k(sJ|j|j |}|jdk(s td|j|}t|||}|j|j|}|j|||dS)z, Take items along any axis. rLzindexer should be 1-dimensional)verifyT)rrrar)rrkrrMrjrrbr} ValueErrorr\r$r=rr)rErrarrres rHrzBaseArrayManager.take$s'2::.=W =.}}'66'##D)||q >? ?   d #'6BZZ%**73 $$t%  rJc|r|Jt|jdS|tj}t |\}}t ||jdd|}|Sr^)NullArrayProxyr\rkrr r3)rErrrj array_valuess rHrzBaseArrayManager._make_na_array=sf % %%!$"3"3A"67 7  J3J?z$UD,=,=bq,A:N rJcpt|j|jD]\}}t||ryy)z Used in .equals defined in base class. Only check the column values assuming shape and indexes have already been checked. FT)zipr>r)rEotherleftrights rH _equal_valueszBaseArrayManager._equal_valuesIs3 t{{ELL9KD%e,:rJTr>r?rFr@rGrreturnNonerBrr9rr&rr@)rztuple[int, ...])rar6rint)rar6rer&rr)rznpt.NDArray[np.object_])rqr<rrrr)rstr)rzlist[str] | Nonerr9)rrrr9)rr rr9)Fraise)r bool | Nonerr rr9)rrrr9)rr rr9rr)rrrr9F)rrrr9)rzbool | Literal['all'] | Nonerr9)NFTFF) rar6rrrrrrrrrr9)NFTF) rznpt.NDArray[np.intp] | Nonerar6rrrrrrrr9)rLT)rznpt.NDArray[np.intp]rar6rrrr9)NF)rr)'r| __module__ __qualname____doc__ __slots____annotations__rIrNpropertyrSrFr\ staticmethodrbrfrorrrurxrrrrrrrr(rrrrrrrrrrrrr_rJrHr<r<esc" I .-  "& "1"" "  "(.. 33 & L '(,63%63  63p95vN+T $,DIRV  =@     %% D J  "HR   "          :  "MH-MH MH  MHMHMH MHd  %      2 rJr<cLeZdZeddZ d ddZddZddZdddZd dZ d!dZ ed"dZ d# d$d Z d% d&d Z d'd(d Zd)dZd*dZd*dZd+dZd dd d,dZd)dZd d ej,f d-dZed.dZed.dZy )/ ArrayManagercy)Nrzr_rRs rHr}zArrayManager.ndimXrJc||_||_|rj|Dcgc] }t|c}|_|Dcgc]}t|ddd}}|Dcgc] }t |c}|_|j yycc}wcc}wcc}wNrLr)r=r>r'r0r1_verify_integrity)rEr>rFrGrYrrns rHrIzArrayManager.__init__\s  59:Tr,r*T:DJCIJ6a*1dA6q96FJ?EFv.s3vFDK  " " $ :JFsA<BBc|j\}}t|j|k(s%tdt|jd|d|jD]}t||k(stdt|d|dt |t j tfstdt|d|jd k(rttd |jd y) NzANumber of passed arrays must equal the size of the column Index: z arrays vs z columns.z=Passed arrays should have the same length as the rows Index: z vs z rowszDPassed arrays should be np.ndarray or ExtensionArray instances, got z insteadrLz6Passed arrays should be 1-dimensional, got array with  dimensions instead.) r\rWr>rrrkrrrMr})rEn_rows n_columnsrns rHrzArrayManager._verify_integrityms -- 4;;9,St{{#$K {)E ;;Cs8v% S3xjVHE3cBJJ#?@ 9+X/88q= Lxxj 46rJc$t|jDcgc]}|jc}}|jDcgc]}|| }}t|tr"|j j ||}nqt|r"tj ||j}nDt|r"tj ||j}ntj||}t|g|jdgScc}wcc}w)z Return the array corresponding to `frame.iloc[loc]`. Parameters ---------- loc : int Returns ------- np.ndarray or ExtensionArray rirL)r,r>rjrrconstruct_array_type_from_sequencerr_ndarrayrr rkrlSingleArrayManagerr=)rElocrnrjrrs rHfast_xszArrayManager.fast_xss" "D 399 "DE&*kk2ks#c(k2 e^ ,//1@@u@UF #E *"11&FOOF $U +#226GPPFXXfE2F!6(TZZ]O<<#E2s D D c|j|}|dk(r|jDcgc]}|| }}n|dk(r|j|}t|j}||j |||<t ||dScc}w)NrrLFr)rbr>rr=_getitem_slicerM)rEslobjrarnr>rs rH get_slicezArrayManager.get_slices##D) 19,0KK8KSc%jKF8 QY[['F #!$66u=tDz&(UCC9s BcV|j|}t|g|jdgS)z: Return the data as a SingleArrayManager. r)r>r'r=)rErrs rHigetzArrayManager.igets*Q!6(TZZ]O<)rErs rH iget_valueszArrayManager.iget_valuess{{1~rJcf|jDcgc]}tj|c}Scc}w)zB Used in the JSON C code to access column arrays. )r>rkrrms rH column_arrayszArrayManager.column_arrayss( ,0;;7;C 3;777s.FNctj|rt|tjr,|j dk(r|j ddk(sJ|dddf}t|}t|tjtfsJ|j dk(sJt|t|jdk(sJ||j|<yt|trct|j |jnd|j |jn|j d|j" |j"nd}nEt|tjsJ|j$dk(sJtj&|d}|j dk(sJ|j dt|jdk(sJt)|D]\}}|dd|f}||j|<y)a Set new column(s). This changes the ArrayManager in-place, but replaces (an) existing column(s), not changing column values in-place). Parameters ---------- loc : integer, slice or boolean mask Positional location (already bounds checked) value : np.ndarray or ExtensionArray inplace : bool, default False Whether overwrite existing array as opposed to replacing it. rzrLNrr)rrrrkrr}shaper1rrWr=r>rrangestartstopr\steprjnonzeror) rEr(rinplacerefsr value_idxmgr_idx value_arrs rHisetzArrayManager.isets, >># %,q{{1~***ad (.Eebjj.%AB BB::? "?u:TZZ]!33 33$DKK  U #*/ YY2 HH0d6G6G6JHH0a+G c2::. ..99& &&jjoa(GzzQ{{1~TZZ]!3333"+G"4 Iwal+I#,DKK #5 rJc t|s td|j|}t|g|jdg}|r|j ||y|j |f|}|jd|j|<y)z Set values ("setitem") into a single column (not setting the full column). This is a method on the ArrayManager level, to avoid creating an intermediate Series at the DataFrame level (`s = df[loc]; s[idx] = value`) z%The column index should be an integerrN)r TypeErrorr>r'r=setitem_inplacer)rEr(idxr inplace_onlyrnrqnew_mgrs rHcolumn_setitemzArrayManager.column_setitemsy#CD Dkk# #A8    U +kk3&%0G&~~a0DKK rJcv|jj||}t|d}|jdk(r4|jddk(r |dddf}nt d|jt |}|jj}|j||||_||jd<y)z Insert item at selected position. Parameters ---------- loc : int item : hashable value : np.ndarray or ExtensionArray T) extract_numpyrzrrLNz-Expected a 1D array, got an array with shape ) rSinsertr"r}r6rr1r>rr=)rEr(itemrr=rr>s rHrKzArrayManager.inserts::$$S$/e48 ::?{{1~"ad  CEKK=Q$E* !!# c5!   1 rJc:tj|jdtj}d||<tj|dDcgc]}|j |c}|_|j d|j d|g|_|Scc}w)z] Delete selected locations in-place (new block and array, same BlockManager) rriFrL)rkonesr6bool_r;r>r=)rErto_keeprs rHideletezArrayManager.idelete8s''$**Q-rxx8 /1zz'/B1/EF/E!t{{1~/EF jjmTZZ]7%;<  GsBcg}g}t|jD]d\}}t|d}||}|jdk(r|jddk(sJ|d}|j ||j |ft |dk(rd}n|djd}tt|}|j} t|||| gS)z Apply grouped reduction function columnwise, returning a new ArrayManager. Parameters ---------- func : grouped reduction function Returns ------- ArrayManager rz)r}rrL) rr>r.r}r6rrWr&r7rSrM) rErrresult_indicesrrnresnrowsindexcolumnss rHgrouped_reducezArrayManager.grouped_reduceFs+- $& ,FAs$Sq1Cs)Cxx1}yy|q(((!f   %  ! !! $- }  "E!!$**1-EeEl#**tDz-%)9::rJc^g}t|jD]C\}}||d}|tur |jnd}|j t |gd|Et jtjdgt}|j}t||||g} | S)z Apply reduction function column-wise, returning a single-row ArrayManager. Parameters ---------- func : reduction function Returns ------- ArrayManager rr`Nri) rr>rrjrr#r& _simple_newrkrlrhrSrM) rErrrrnrTrjrVrWrGs rHreducezArrayManager.reducems+-  ,FAss#C "%CIIE  ud%8 -!!"((D6"@A**$t*]UG,<=rJc|j}|j}t||Dcgc]\}}|||}}}t|||jScc}}w)zO Apply array_op blockwise with another (aligned) BlockManager. )r>rrMr=)rErarray_op left_arrays right_arraysrrrs rHoperate_blockwisezArrayManager.operate_blockwisesc kk || 58l5S 5SkdEHT5 !5S  tDz-44 sAlinear) transposed interpolationc |jDcgc]}t|d}}|Dcgc],}t|tj|j |.}}t |D]@\}}|jdk(s|jddk(sJ|j|d||<B||jdg} t||| Scc}wcc}w)NrzrrL) r>r.rrkrrrr}r6r=rM) rEqsrbrcrarrsnew_arrsrrnrFs rHquantilezArrayManager.quantiles37++>+Q"1a(+>OS OS!OArzz"**5} Et   )FAsxx1}yy|q(3#))3(!!f * DJJqM"tDz(D))? s C 1Cc |j\}}|jjr |}d}d}d}nqtj|jj d}|||j<d}|jj |j}|jd}|j |j} t| } g} |jD]i} t|jdD]L} |r"t| | dd| f|| ||dd| f} nt| | dd| fd } | j| Nk|j}|j|j d}||g}t#|| |d S) a' Return a BlockManager with all blocks unstacked. Parameters ---------- unstacker : reshape._Unstacker fill_value : Any fill_value for newly introduced missing values. Returns ------- unstacked : BlockManager FNrPTrr`rLrrr)_indexer_and_to_sortrrrkfullr6reshape full_shaperrr>r7rr new_indexget_new_columnsr=rM)rE unstackerrr_ new_indexerr new_mask2Dr new_indexer2Drrnrnew_arrro new_columnsrs rHunstackzArrayManager.unstacks33  >>   !KJJ M'').."6"6;K*1K  'J39>>/22I4H4HIJ&NNN2M+ ++Y-A-AB +M:  ;;C9//23%%ad+#0#3#-'1- G&c=A+>5QG!!'*4'' // 1 > {+tDz*hGGrJc>t|jdk(r5tj|jt }|j S|xs|tju}|s-t|jDcgc]}|jc}}t|}tj|j|}t|jD]!\}}|j||}||dd|f<#|tjur||t|<|Scc}w)a# Convert the blockmanager data into an numpy array. Parameters ---------- dtype : object, default None Data type of the return array. copy : bool, default False If True then guarantee that a copy is returned. A value of False does not guarantee that the underlying data is not copied. na_value : object, default lib.no_default Value to be used as the missing value sentinel. Returns ------- arr : ndarray rrirN)rWr>rkemptyr6float transposer no_defaultr,rjr+r\rrr)rErjrna_value empty_arrrnrrs rHas_arrayzArrayManager.as_arrays0 t{{ q 59I&&( (5xs~~5%DKK&HKSsyyK&HIE&$++59 ,FAs**U*.CF1a4L- 3>> )#+F4<  'Is9Dcttjj|Dcgc]}|jc}}|||d|dgd}|Scc}w)zK Concatenate uniformly-indexed ArrayManagers horizontally. rLrFr)r itertoolschain from_iterabler>)clsmgrsrFrqr>rGs rHconcat_horizontalzArrayManager.concat_horizontalsV ioo3344P4CSZZ44PQRftAwQ05I5QsA c tt|djDcgc]=}ttt|Dcgc]}||j|c}?}}}|||d|dgd}|Scc}wcc}}w)zI Concatenate uniformly-indexed ArrayManagers vertically. rrLFr)r7rWr> concat_arrays)rrrFjrr>rGs rHconcat_verticalzArrayManager.concat_verticals3tAw~~./ / eCI6FG6F47>>!,6FG H/  ftAwQ05I H s BA? B?B)rz Literal[2]rrr r(r rr'r)r,rrar6rr)rr rr')rr rr5)rzlist[ArrayLike])FN)r(int | slice | np.ndarrayrr5r<rrrr)r(r rErrFrrrrB)r(r rLr4rr5rr)rr)rrrr9)rrrr)rer&rbrrcr8rr)rrrrhrz np.ndarray)rz list[Self]rFr@rr9)r|rrrr}rIrr)r-r/r2r4rArHrKrQrXr[r`rhrxrr~r classmethodrrr_rJrHrrWs "& %1%% %  %"6=4 D= 88  ; %;; ;  ;|TY1151LP1 1(!!F %;N@ 5 !/7 * * * - *  **3Hn>> .. .  .`  rJrc@eZdZUddgZded<ded<eddZ d ddZd dZe dZ d!d"d Z e d#d Z ed$d Zed%d ZedZdZdZdZed&dZed&dZd'dZd(d)dZd*dZd"dZdd+dZd#dZd,dZd-dZd.dZy )/r'r=r>r?r@cyrr_rRs rHr}zSingleArrayManager.ndim5rrJc0||_||_|rt|dk(sJt|dk(sJ|jDcgc] }t|c}|_|d}t |}t |ddd}|g|_|j yycc}wr)r=r>rWr'r1r0r)rEr>rFrGrYrns rHrIzSingleArrayManager.__init__9s   t9> !>v;!# ##59ZZ@Zr,r*Z@DJ)C%c*C&sD!4Q7C%DK  " " $ AsBc|j\}t|jdk(sJ|jd}t||k(sJ|jdk(st d|jdy)NrLrz5Passed array should be 1-dimensional, got array with r )r6rWr>r}r)rEr!rns rHrz$SingleArrayManager._verify_integrityLstJJ 4;;1$$$kk!n3x6!!!xx1}G88*02 rJc|SrBr_r`s rHrbz"SingleArrayManager._normalize_axisWs rJNc|tgtg}tjg|j}t ||g|S)z9Return an empty ArrayManager with index/array of length 0ri)r&rhrkrlrjrM)rErFrls rHrNzSingleArrayManager.make_empty[s? <"F+,DHHRtzz:tDz5'4((rJc||g|gSrBr_)rrlrVs rH from_arrayzSingleArrayManager.from_arraybsE7UG$$rJc|jSrBrQrRs rHrFzSingleArrayManager.axesg zzrJc |jdS)NrrQrRs rHrVzSingleArrayManager.indexkszz!}rJc.|jjSrB)rlrjrRs rHrjzSingleArrayManager.dtypeoszzrJc,t|jS)z$The array that Series.values returns)r/rlrRs rHr/z"SingleArrayManager.external_valuessstzz**rJc|jS)z%The array that Series._values returns)rlrRs rHinternal_valuesz"SingleArrayManager.internal_valueswrrJch|j}t|tjr t |}|S)z#The array that Series.array returns)rlrrkrrrms rHrzSingleArrayManager.array_values{s(jj c2:: &%c*C rJct|jtjr"|jjj dvS|jj S)Niub)rrlrkrrjkind _can_hold_narRs rHrzSingleArrayManager._can_hold_nasA djj"** -::##((5 5::** *rJcy)NTr_rRs rHrz"SingleArrayManager.is_single_blocksrJctd)NzUse series._values[loc] insteadrC)rEr(s rHr)zSingleArrayManager.fast_xss!"CDDrJc||jk\r td|j|}|jj |}t ||g|gdS)NrFr)r}rrlrVr+rM)rEr,ra new_arrayros rHr-zSingleArrayManager.get_slicesU 499 BC CJJu% JJ--e4 tDz9+ {UKKrJcf|j|}|j|}t||g|gSrB)rlrVrM)rErrros rHget_rows_with_maskz%SingleArrayManager.get_rows_with_masks5JJw' JJw' tDz9+ {33rJc t|r||jfi|}nt|j|di|}t||g|jS)Nr_)rrlrrMr=)rErrrs rHrzSingleArrayManager.applysO D>TZZ262I1 D1;F;ItDz9+tzz22rJct|tjr1|j|jkDrt d|j|j d||S)z Set values with indexer. For SingleArrayManager, this backs s[indexer] = value See `setitem_inplace` for a version that works inplace and doesn't return a new Manager. zCannot set values with ndim > rr)rrkrr}rrrs rHrzSingleArrayManager.setitemsP grzz *w||dii/G=dii[IJ J$$Yu$MMrJctj|jdtj}d||<|jd|g|_|j d|g|_|S)zS Delete selected locations in-place (new array, same ArrayManager) rriF)rkrNr6rOr>r=)rErrPs rHrQzSingleArrayManager.ideletes]''$**Q-rxx8 {{1~g./ jjmG,-  rJc||jr(t||j|jdS|j S)NFr)rlrMr>r=rN)rErs rHrz#SingleArrayManager._get_data_subsets8 TZZ 4:dkk4::N N??$ $rJc"||jd<y)z Set (replace) the values of the SingleArrayManager in place. Use at your own risk! This does not check if the passed values are valid for the current SingleArrayManager (length, dtype, etc). rNr1)rErs rH set_valueszSingleArrayManager.set_valuess  ArJc`|jdg}|jd|g}t||dS)z5 Manager analogue of Series.to_frame rFr)r>rFr)rErWr>rFs rH to_2d_mgrzSingleArrayManager.to_2d_mgrs5++a.! ! g&FD5AArJ)rz Literal[1]rrr rBr)rr'r rrrr)r,rrar6rr')rznpt.NDArray[np.bool_]rr')rrrr')rrrr')rr5rr)rWr&rr)r|rrrrrr}rIrrrbrNrrrFrVrjr/rrrrr)r-rrrrQrrrr_rJrHr'r',s5I .-  "& %1%% %  %& )%%  +++EL4 3 N % BrJr'c6eZdZdZdZddZeddZd dZy) rab Proxy object for an all-NA array. Only stores the length of the array, and not the dtype. The dtype will only be known when actually concatenating (after determining the common dtype, for which this proxy is ignored). Using this object avoids that the internals/concat.py needs to determine the proper dtype and array type. rLc||_yrBrrs rHrIzNullArrayProxy.__init__s rJc|jfSrBrrRs rHr6zNullArrayProxy.shapesyrJct|trd|jjg|}t j |j tj }|j|dSt|}t|}t j|j |}|j|t|S)a' Helper function to create the actual all-NA array from the NullArrayProxy object. Parameters ---------- arr : NullArrayProxy dtype : the dtype for the resulting array Returns ------- np.ndarray or ExtensionArray riTrj)rrr$r%rkrNrrrr rr{fillr!)rErjr{rrrns rHto_arrayzNullArrayProxy.to_arrays e^ ,..0??%?PEwwtvvRWW55G::g$:7 7-U3E+E2J((466/C HHZ 1#6 6rJN)rr rr)rz tuple[int])rjr7rr5) r|rrrr}rIrr6rr_rJrHrrs, D7rJrc|Dcgc]}t|tr|}}|Dchc]}|j}}t|dk(}|r|dj}n>t d|Dr t |}n#t |Dcgc]}|jc}}|Dcgc]1}t|tr|j|n t||d3}}t|dtrt|d}|j|Stj|}t|dk(rJ|D chc]} | jj} } t| dk7rd| vr|jt }|Scc}wcc}wcc}wcc}wcc} w)a Alternative for concat_compat but specialized for use in the ArrayManager. Differences: only deals with 1D arrays (no axis keyword), assumes ensure_wrapped_if_datetimelike and does not skip empty arrays to determine the dtype. In addition ensures that all NullArrayProxies get replaced with actual arrays. Parameters ---------- to_concat : list of arrays Returns ------- np.ndarray or ExtensionArray rLrc3HK|]}tj|dyw)rN)r is_np_dtype)rXrs rHrZz concat_arrays..!s 71S__Q &s "Frzb)rrrjrWrrr rr rrM_concat_same_typerk concatenaterrrh) to_concatrto_concat_no_proxydtypes single_dtype target_dtypernrrrkindss rHrrs&&/TYjN6S!YT1 21!agg1F 2v;!#L)!,22 7 7 7*F3 '>P(Q>Ps>P(QR  C c> * \" #|% 8 9 )A,/9Q< $$Y// ^^I &F 6{a+=>+=C+=> u:?e|v. MEU 2)R"?s!E5E5E:>E?6F-F )rrrr5)Zr __future__rrtypingrrrnumpyrk pandas._libsrrpandas.core.dtypes.astyper r pandas.core.dtypes.castr r r rpandas.core.dtypes.commonrrrrrrpandas.core.dtypes.dtypesrpandas.core.dtypes.genericrrpandas.core.dtypes.missingrrrpandas.core.algorithmscore algorithmsr pandas.core.array_algos.quantilerpandas.core.array_algos.takerpandas.core.arraysrrrr pandas.core.constructionr!r"r#pandas.core.indexersr$r%pandas.core.indexes.apir&r'pandas.core.indexes.baser(pandas.core.internals.baser)r*r+r,pandas.core.internals.blocksr-r.r/r0r1r2pandas.core.internals.managersr3collections.abcr4pandas._typingr5r6r7r8r9r:r<rr'rrr_rJrHrs#   5 '&<0  8 9(l{ldR#RjiB)+<iBX,7,7^5rJ