ehe ddlmZddlZddlmZmZddlZddlZddl m cm Z ddl mZddlmZddlmZddlmZmZddlmZmZmZmZdd lmZdd lmZddlm cm!Z"dd lm#Z#m$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7m8Z8erddl9m:Z:m;Z;mZ>ddl?m@Z@GddZA d" d#dZBd"d$dZC d" d%dZD d&dZEd'd(dZFd)d(dZGd*dZH d' d+dZI d,d ZJd-d!ZKy).) annotationsN) TYPE_CHECKINGcast)PerformanceWarning)cache_readonly)find_stack_level)find_common_type maybe_promote)ensure_platform_intis_1d_only_ea_dtype is_integerneeds_i8_conversion)ExtensionDtype)notna) factorizeunique)factorize_from_iterable)ensure_wrapped_if_datetimelike) DataFrame)Index MultiIndex RangeIndex)concat)Series)compress_group_indexdecons_obs_group_idsget_compressed_idsget_group_indexget_group_index_sorter) ArrayLikeLevelnpt)ExtensionArray) FrozenListceZdZdZ d ddZe ddZeddZddZdZ eddZ eddZ dd Z dd Z dd Zedd ZeddZy ) _Unstackera  Helper class to unstack data / pivot with multi-level index Parameters ---------- index : MultiIndex level : int or str, default last level Level to "unstack". Accepts a name for the level. fill_value : scalar, optional Default value to fill in missing values if subgroups do not have the same set of labels. By default, missing values will be replaced with the default fill value for that data type, NaN for float, NaT for datetimelike, etc. For integer types, by default data will converted to float and missing values will be set to NaN. constructor : object Pandas ``DataFrame`` or subclass used to create unstacked response. If None, DataFrame will be used. Examples -------- >>> index = pd.MultiIndex.from_tuples([('one', 'a'), ('one', 'b'), ... ('two', 'a'), ('two', 'b')]) >>> s = pd.Series(np.arange(1, 5, dtype=np.int64), index=index) >>> s one a 1 b 2 two a 3 b 4 dtype: int64 >>> s.unstack(level=-1) a b one 1 2 two 3 4 >>> s.unstack(level=0) one two a 1 3 b 2 4 Returns ------- unstacked : DataFrame c||_||_|j|_|jj ||_d|jj |j vrdnd|_t|jj|_ t|jj|_ |jj|j |_|jj|j |_|j|j |_|jslt#|jj |j }|jj%||_|j j%||_t'j(|jDcgc]}|j*c}}|jj*}||z} | t'j,t&j.j(kDr(t1j2d| dt4t7|j9ycc}w)Nrz%The following operation may generate z& cells in the resulting pandas object.) stacklevel) constructorsortremove_unused_levelsindex_get_level_numberlevelcodesliftlistlevelsnew_index_levelsnamesnew_index_namespop removed_name removed_levelremoved_level_fullrtakenpmaxsizeiinfoint32warningswarnrr_make_selectors) selfr.r0r+r, unique_codes index_levelnum_rows num_columns num_cellss Y/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/reshape/reshape.py__init__z_Unstacker.__init__ns' //1 ZZ11%8 tzz// ;;A !%TZZ%6%6 7#DJJ$4$45 0044TZZ@!2266tzzB"',,tzz":yy!$**"2"24::">?L!%!3!3!8!8!FD &*&=&=&B&B<&PD # 66t?T?TU ;++UV((-- {*  rxx)-- - MM7 {C23"+-   VsIc^|j}t|jj}t|jj}|d|||dzdz||gz}t d|d|||dzdz||gzD}t ||\}}t|}t||} | |fS)Nr)c32K|]}t|ywNlen.0xs rK z2_Unstacker._indexer_and_to_sort..sKc!fK) r0r3r.r1r4tuplerrQr) rEvr1levsto_sortsizes comp_indexobs_idsngroupsindexers rK_indexer_and_to_sortz_Unstacker._indexer_and_to_sorts JJTZZ%%&DJJ%%&)eAEGn,az9Kd2Aha!eg&>$q'&JKK0%@ Gg,(W=c|j\}}|jr|Dcgc]}|j|c}S|Scc}wrO)r`r,r<)rEr_rZlines rK sorted_labelsz_Unstacker.sorted_labelss=44 993:;4DIIg&; ;.s7qCF7rVdtypeTz0Index contains duplicate entries, cannot reshape) return_indexr))r5rdrWrrQr r.levshaper0r2 full_shaper=zerosprodboolputsum ValueError group_indexmaskr, searchsortedarange compressorr) rE new_levelsremaining_labels level_sizesr\r]r^strideselectorr{s rKrDz_Unstacker._make_selectorssD**  --cr27J77 01A;O Gg,(4 $$TZZ0499<!6/%%b)FZ,??$))Kxx0= 4 88:DJJ 'OP P% 99(55bii6HIDO ggbii &Nq&QRDOracHt|jjSrO)rvr{all)rEs rKmask_allz_Unstacker.mask_allsDIIMMO$$ractjt|jtj}|j |d\}}||j dfS)Nror( fill_valuer)r=r}rQr.intpget_new_valuesany)rE dummy_arr new_valuesr{s rK arange_resultz_Unstacker.arange_resultsMIIc$**oRWW= ..yR.H D488A;&&rac.|jdk(r|ddtjf}||jddk7r t d|j ||\}}|j |}|j}|j||||jS)Nr)z-must pass column labels for multi-column data)r.columnsrp) ndimr=newaxisshaperyrget_new_columns new_indexr+rp)rEri value_columnsrrjrr.s rK get_resultz_Unstacker.get_results ;;! ArzzM*F  V\\!_%9LM M'' ; &&}5 %    raNc |jdk(r|ddtjf}|j|}|j\}}|j d}||z}||f}|j } |j} | r\t|rQ|j|||jddj|} tj|t} | | fS|j} | r$|j} tj|| } npt| t r)| j#}|j%|| } || ddn7t'| |\} }tj|| } | j)|| j*}tj,|t} t/|jr#|j1d}| j1d} n|j3|d}t5j6|| j1d|||| | j1dt/|jr7| j1d} t9| } | j1|j} | | fS) Nr)roi8Fcopyu1zM8[ns])rr=rrlrsrr{rrQreshapeswapaxesonesrvrpempty isinstancerconstruct_array_type_emptyr fillnamertrviewastype libreshapeunstackr)rErirrklengthwidthr result_width result_shaper{rrnew_maskrpclsrs rKrz_Unstacker.get_new_valuess% ;;! ArzzM*F008  av~  - yy== F %%feV<!Q&  ww|48Hx' '  LLE,e->?I# I>> 512YiRW  Ns,F0cxt|jt|jk7rJ|jj|j}|jrt j |dd}|St|j|jz}t j||jz }|S)Nrr()rQr;r: get_indexerr2r=rr})rErrs rKrz_Unstacker._repeaterks t&& '3t/A/A+B B..::4;M;MNHyy99Xq"5 ++,tyy8Fyy(4994Hrac|jddDcgc]}|j|j}}t|jdk(ry|jd|d}}|dk(j r%|j t||j}|j|j|jdSt|j||jdScc}w)Nr(r)rFr) rdr<r~rQr5rrrrr7r)rEr result_codesr0 level_codess rKrz_Unstacker.new_indexzs>B=O=OPSQS=TUc1U U t$$ % *!%!6!6q!9<?;Er!&&( SZA::k*11$2F2Fq2IJ J((&&"   Vs"C*)T)r.rr0r!r,rvreturnNone)rz-tuple[npt.NDArray[np.intp], list[np.ndarray]])rzlist[np.ndarray])ri np.ndarrayrr)rrv)rz2tuple[npt.NDArray[np.intp], npt.NDArray[np.bool_]])rrrO)rz Index | None)rr)rr)__name__ __module__ __qualname____doc__rLrr`rdrlrDrrrrrrrrarKr&r&@s+\JN,,(-,BF, ,\    & S8%%''  M$^$ L    rar&ct|dk(r|S|j}tt|}||jvr|g}|Dcgc]}|j |}}t |jDcgc] }||vs| }}|Dcgc]}|j|}}|Dcgc]}|j|}}|Dcgc]}|j|} }|Dcgc]}|j|} }|Dcgc]}|j|} }|Dcgc]}|j|} }td|D} t|| dd}t|d\}}t||| |d}|st|d}nt| |gz| |gz| dgzd }t|t r3|j#}||_|j%d|| }|}| }|}n t|j&trG|}|rA|j)d}|j%||| }|Dcgc]}||kr|n|d z }}|rA|S|j#d }||_|j%d|| }t|t r |jn |j&ttsJjdg|z}|j&j*g| z}jdg}|j-fd |Dt|||d }t|t r ||_|S||_|Scc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}w)Nrc32K|]}t|ywrOrPrRs rKrUz$_unstack_multiple..s*Q#a&*rVF)r,xnullr,)r__placeholder__rrrr,r)deepc3ZK|]"}|jjd$yw)r(N)r<r1)rSrecunstcolss rKrUz$_unstack_multiple..s#N#(.."45Ns(+)rQr.rrr6r/rangenlevelsr4r1rWrrrrrrrrrr8rextend)dataclocsrr,r.irlocsclevelsccodescnamesrlevelsrcodesrnamesrrzcomp_idsr] recons_codes dummy_indexdummy unstackedrrrresultvalrXdummy_df new_columnsrs @rK_unstack_multiplersc 5zQ  JJE U #E  16 7AU $ $Q ' 7E 7emm, ?1Q ?E ?(-.1u||A.G.&+ ,ekk!n ,F ,&+ ,ekk!n ,F ,(-.1u||A.G.&+ ,ekk!n ,F ,&+ ,ekk!n ,F , *'* *E!&%e5IK,[uEHg''5&PUVL G*;<  gY&H:%-.."  $ ! MM"3 QUMV    dllJ /Fiil N:?@Qa#g1q50@@ M99%9($$$ *4%  i ( H ((H(J///ooa()G3 \\&&'&0 ^^A&' NNN)eK)V$%  (  S 8 ?. , ,. , ,BAs<L,4 L1>L1L6$L;MM8M M"Mct|ttfr"t|dk7rt ||||S|d}t |s |dk(s|j j|t|trEt|j trt||||S|jjdSt|j ts"tdt|j dt|j rt#|||| St%|j ||j&| }|j)|j*d| S) Nr)rrrT) future_stackz'index must be a MultiIndex to unstack, z was passedrr0r+r,rr)rrWr3rQrr r.r/rr_unstack_frameTstackrytyper rp_unstack_extension_seriesr&_constructor_expanddimr_values)objr0rr, unstackers rKrrs+%%' u:?%S%JTR R!HE e U.?%? ##E*#y! cii ,!#u$O O55;;D;1 1  : .5d399o5Fk R   syy ),S%$O O IIU0J0JQU ## KKt $  rac\t|jtsJt|j||j|}|j s:|j j||}|j||jS|j|j|j|S)Nrr)axesr) rr.rr& _constructor_can_fast_transpose_mgrr_constructor_from_mgrrrrr)rr0rr,rmgrs rKrr s cii ,, , C,<,<4I  " "hhyZ@((388(<<## KKs{{z$  rac|j}|j|||}|jjdg|_|S)an Unstack an ExtensionArray-backed Series. The ExtensionDtype is preserved. Parameters ---------- series : Series A Series with an ExtensionArray for values level : Any The level name or number. fill_value : Any The user-level (not physical storage) fill value to use for missing values introduced by the reshape. Passed to ``series.values.take``. sort : bool Whether to sort the resulting MuliIndex levels Returns ------- DataFrame Each column of the DataFrame will have the same dtype as the input Series. )r0rr,r)to_framerr_drop_level_numbers)seriesr0rr,dfrs rKrrsC8  B ZZe Z FF^^77.stack_factorizeMs; ??"))CJ// /3E:z5  ra) level_numdropnar,Frrr.)rrr/rr_stack_multi_columnsr.r3r4r1rrr=rravelr6rzipmapr_is_homogeneous_typedtypesrrr_concat_same_typeitems"_reorder_for_extension_array_stackr_constructor_sliced)framer0rr,rNKrrrrclevclabrrr4ilabr1rrparrrjcolrr{s rKrrCs! ;;DAq //6I%--,# YvD   EKK ,%++,,- .3kk.?.?@sSZZ]@ @$U]]3 d$q)//12**+ ++,YiRW  #C%++u}}9U$VW t Aa 0 6 6 88;;##U]]%7%78"   ;;555ell**+q  e^ ,,,.C..+0;;=9C9J:s (LL ctfd|Dr}|D]}t||||}|Std|Drj}|Dcgc]}jj|}}|r<|j d}t||||}|Dcgc]}||kr|n|dz }}|r<|St dcc}wcc}w)Nc3NK|]}|jjvywrO)rr6)rSrrs rKrUz!stack_multiple..s 7#3%--%% % 7s"%)rr,c3<K|]}t|tywrO)rint)rSrs rKrUz!stack_multiple..s 3cZS ! 3srr)zTlevel should contain all level names or all level numbers, not a mixture of the two.)rrrr/r8ry)rr0rr,rrrXs` rKstack_multipler+s 7 77 BC63vDAF B2 M+ 3U 3 3BGG#005GG))A,C63vDAF8==!!s(QA-=E=  M  1  H>s "B7B<c vt|jdkr,|jdj|jdSt |jdd|j ddDcgc]\}}|Dcgc]}|dk\r||ndc}!}}}}t |}dt j|D}t |}tjt ||jDcgc]"\}}d|vrt||jn|$c}}|jddScc}wcc}}}wcc}}w) zBCreates a MultiIndex from the first N-1 levels of this MultiIndex.rrrNr(c3&K|] \}}| ywrOr)rSkeyrjs rKrUz,_stack_multi_column_index..sAVS!SAsror6) rQr4rr6rr1 itertoolsgroupbyr from_arraysrrp) rrr1crYtuples unique_tuplesnew_levsnew_levs rK_stack_multi_column_indexr8s6 7>>a~~a ((gmmA.>(??gnnSb17=="3EF   C.3316Qt #3 D  $ZFAy'8'8'@AMM"H  ! !!$Hgnn = 047/BE' + O mmCR   4  s/ D.;D)D.+'D5 )D.c  dd}|jd}|j}t|tsJ||jdz k7rP|}t ||jdz D])}|||} ||dz|} |j | | }+|x|_}|js*|r(|d|} |j| d}|j}tt|}t|} i} |jd}t|jd}|rtj|}|j!t#|d}tj$||}t#|}g}| D]} |jj'|}t|t,s t#|}n|j.|j0z }||k7rk|j2dd|j|f}|j%|jjd|_|j5|j6}n|j8dd|f}t;|j<j?}t|t@r|jCjE|jGDcgc]"\}}|jHjK|d $c}}}|jL\}}tjN||zjQ||jRjU}|j%|}n |j6}|jVdkDr|jU}|| |<t#|dkDr| jY|} t#|}t|jZtrqt]|jZj} t]|jZj^}!|jZjD"cgc]}"|"ja|}#}"nDtc|jZ\}$}%|%g} |$ja|g}#|jZjdg}!| j+||#j+tjf|||!j+|jj^|t| |#|!d }&|ji| |&| }'|jjdkDrJ|jjk|gj}(|'jjm|(s|'|(}'|r|'jodd }'|'S#t($r|j+|Y wxYwcc}}wcc}"w)Nc@||jvr|j|S|S)z Logic for converting the level number to something we can safely pass to swaplevel. If `level_num` matches a column name return the name from position `level_num`, otherwise return `level_num`. r/)rrs rK_convert_level_numberz3_stack_multi_columns.._convert_level_numbers%  %==+ +raFrr)r)r0rfr()rrr)r.rr)rfhow)rr*rr)8rrrrrr swaplevel _is_lexsorted sort_indexrr8r4rr1r=r,rrQr<get_locKeyErrorrslicestopstartlocreindexriilocr rtolistrrrrrrrr}rrrr differencer.r3r6rrrrrr equalsr))rrrr,r;thismi_cols roll_columnsrlev1lev2 level_to_sortrnew_data level_valsrlevel_vals_nanlevel_vals_usedlevsize drop_colsr.rE slice_lenchunk value_slicesubsetrprjrTr r!idxrrrr old_codes old_levelsrrdesired_columnss) rKrrs  ::5: !DllG gz ** *GOOa'' y'//A"56 >A(L9D(Q =D'11$=L  > ".- w  "t.a9 ];,,:w'G+G4KH#Jr*+K ggk* &&s:=Nggnk:O+GI'$ ,,&&s+C#u%CI399,I  HHQ S 112E*// 0C0CB0GHEM---@GGKYYq#v&F$V]]%9%9%;)..s3 $mm   a %++-K# O'$R 9~!,,Y7  D A$**j)$**++, ))* 48JJ4D4DESSZZ(E E 7 C : \ %%g./ ZZ__% j! RWW[!,- U]]((34)eI   ;  OF }}q --;;YKHOOQ~~$$_5O,FA51 MS    S !  0U,Fs1U#'U; VU87U8ctj||zj||jj }|j |S)a Re-orders the values when stacking multiple extension-arrays. The indirect stacking method used for EAs requires a followup take to get the order correct. Parameters ---------- arr : ExtensionArray n_rows, n_columns : int The number of rows and columns in the original DataFrame. Returns ------- taken : ExtensionArray The original `arr` with elements re-ordered appropriately Examples -------- >>> arr = np.array(['a', 'b', 'c', 'd', 'e', 'f']) >>> _reorder_for_extension_array_stack(arr, 2, 3) array(['a', 'c', 'e', 'b', 'd', 'f'], dtype='>> _reorder_for_extension_array_stack(arr, 3, 2) array(['a', 'd', 'b', 'e', 'c', 'f'], dtype='|j6j@d|f}n>tC|j6d \}}|g}t=tj>|d|f}t5|t8r'|j:}|jEj@}n"|jg}tC|d dg}|Dcgc]!}tjF|t|#}}t9||z||z|j6jHt=|jHzd | _t|}t|}tjJ|}tj>||z|tjFtjJ||z}| jM|} | j d k(rd|jjtk(rBt| jdk(rtO| j6 } n| jPdddf} | j dk(rd| _| Scc}wcc}w)Nz8Columns with duplicate values are not supported in stackT)reverser(r)c3RK|]}|vr tn td ywrO)nextrB)rSkgenr0s rKrUzstack_v3..s,#%ZS U4[8#s$'r)rrpF)use_na_sentinelrrr))rnuniquerQrysortedr rrr=argsort_reorder_ilevelsrriterrWrErrrrrrrrrprJrr.rr4r3rr1rdrop_duplicatesrr6r}r<rrG)rr0 drop_levnumsrf stack_colssorterordered_stack_colsstack_cols_uniqueordered_stack_cols_uniquebufr[rcolumn_indexerrratiorr^ index_levels index_codesr1uniques column_levels column_codeslen_df n_uniquesr_idxsrgs ` @rKstack_v3rus }}#emm"44STT%.L22%--//0CqAUNCDbDIJ 5zA~E"'88@'"))+ 2 9 9 ; C  u}}  "::JKUBIIeSZ0KLKm+L(kk$'9'?'?"@@ FLZF-.Iii "G 776G#V ,ryy69JI/V VD [[ F{{aEMM11SZ? v~~ ! #&,,/F[[A&F {{a MC DRLs V6 V6&V;)NT)rSeries | DataFramer,rv)rrr,rv)rrr,rvrr)r rr,rvrr)r(TT)rrrrvr,rv)TT)rrrr) rrrr*rrvr,rvrr)r%r#r`r*rar*rr#)rrr0z list[int]rr)L __future__rr0typingrrrBnumpyr=pandas._libs.reshape_libsrr pandas.errorsrpandas.util._decoratorsrpandas.util._exceptionsrpandas.core.dtypes.castr r pandas.core.dtypes.commonr r r rpandas.core.dtypes.dtypesrpandas.core.dtypes.missingrpandas.core.algorithmscore algorithmsrgrrpandas.core.arrays.categoricalrpandas.core.constructionrpandas.core.framerpandas.core.indexes.apirrrpandas.core.reshape.concatrpandas.core.seriesrpandas.core.sortingrrrrrpandas._typingr r!r"pandas.core.arraysr#pandas.core.indexes.frozenr$r&rrrrrr+r8rrrrrarKrsu")),24 5,&&CC' .% 25K K ^ DHX X<@Xv! J:>  26  "" "-1""JJBZB8NRA A!$A37AFJAAH" "!$"14""Jhra