eh(ddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$erddl%m&Z&ejNdZ(gdZ) d ddZ*ddZ+ d ddZ,d dZ-d!d"dZ.dZ/d#dZ0d$dZ1y)%) annotationsN) TYPE_CHECKINGcast)NaTlib)InvalidIndexError)find_common_type) safe_sort)Index _new_Index ensure_indexensure_index_from_sequencesget_unanimous_names)CategoricalIndex) DatetimeIndex) IntervalIndex) MultiIndex) PeriodIndex) RangeIndex)TimedeltaIndex)AxiszSorting because non-concatenation axis is not aligned. A future version of pandas will change to not sort by default. To accept the future behavior, pass 'sort=False'. To retain the current behavior and silence the warning, pass 'sort=True'. )r rrrrrrrrr rr rget_objs_combined_axis union_indexesrall_indexes_same default_indexsafe_sort_indexcf|Dcgc]}|j|}}t||||Scc}w)a Extract combined index: return intersection or union (depending on the value of "intersect") of indexes on given axis, or None if all objects lack indexes (e.g. they are numpy arrays). Parameters ---------- objs : list Series or DataFrame objects, may be mix of the two. intersect : bool, default False If True, calculate the intersection between indexes. Otherwise, calculate the union. axis : {0 or 'index', 1 or 'outer'}, default 0 The axis to extract indexes from. sort : bool, default True Whether the result index should come out sorted or not. copy : bool, default False If True, return a copy of the combined index. Returns ------- Index ) intersectsortcopy) _get_axis_get_combined_index)objsraxisrr obj obs_idxess U/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/indexes/api.pyrrHs6<155t$5I5 yIDt TT6s.ct}g}|D];}t||vs|jt||j|=|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r#idsresr%s r'_get_distinct_objsr/jsM EC C c7#  GGBsG  JJsO Jc,t|}t|dk(r tg}nOt|dk(r|d}n;|r!|d}|ddD]}|j|}nt |d}t |}|r t |}|r|j}|S)a Return the union or intersection of indexes. Parameters ---------- indexes : list of Index or list objects When intersect=True, do not accept list of lists. intersect : bool, default False If True, calculate the intersection between indexes. Otherwise, calculate the union. sort : bool, default False Whether the result index should come out sorted or not. copy : bool, default False If True, return a copy of the combined index. Returns ------- Index rNFr)r/lenr intersectionrr rr )indexesrrr indexothers r'r"r"xs4!)G 7|qb  W    QR[ .E&&u-E .gE2U# &   Lr0c\|jr|S t|}t|tr|St t j |}t|tr#tj||j}|St||j|j}|S#t$rY|SwxYw)z Returns the sorted index We keep the dtypes and the name attributes. Parameters ---------- index : an Index Returns ------- Index )names)namedtype) is_monotonic_increasingr isinstancer rnpndarrayr from_tuplesr:r;r< TypeError)r7 array_sorteds r'rrs $$  L '  lE * BJJ 5 eZ (**r ).0inds r' z9union_indexes.._unique_indices..s6#z#u%6sFr rr2NcFt|tr|j}|SrG)r>r tolist)is r'convz4union_indexes.._unique_indices..convs!U#HHJHr0r3)r<) allastypeuniquer,get_indexer_forr4 sort_valuesr rfast_unique_multiple_list) indsr<rIresultr8diffrQrPrs r'_unique_indicesz&union_indexes.._unique_indicess 66 6=ABcCJJu5J1BDB!W^^%FGNN48,E//6"<=D4yt{{}5++-M    ) )D*Aq47*A M  C +Bs C,C1 cDcgc]}t|ts|j!}}|r t|}|Sd}|Scc}w)a Finds a common type for the indexes to pass through to resulting index. Parameters ---------- inds: list of Index or list objects Returns ------- The common type or None if no indexes were given N)r>r r<r )rXidxdtypesr<r6s r'_find_common_index_dtypez/union_indexes.._find_common_index_dtype sK(/I*S%2H#))II $V,E E Js??specialNz0Cannot join tz-naive with tz-aware DatetimeIndexTFrKr3arrayc3@K|]}j|ywrGequals)rHr8r7s r'rJz union_indexes..>s@55<<&@)returnr )r4AssertionErrorr>listr sorted_sanitize_and_checkrtzrBrSobjectunionrRrr;rename) r6rrYkindr[r_xdtisdti_tzsr8r<r;r7s `` @r'rrs  7|qBCC 7|q fd #v vf~. '0MGT  D( y"CajM&BCC"7add&6177 w<3t9~ - NO O t9G $DQZF Y]D>EEqxxUx3EGQZFQR[ GE\\%dd\FF G (1 @GABK@@#GU3E"G,Q/ 5:: LL&E (1w..OD7&Fs G"G,G>GGc\t|Dchc] }t|c}}t|vr[t|dkDrI|Dcgc](}t|tst t|n|*}}|j tn|dfSt|dkDst|vr|dfS|dfScc}wcc}w)aN Verify the type of indexes and convert lists to Index. Cases: - [list, list, ...]: Return ([list, list, ...], 'list') - [list, Index, ...]: Return _sanitize_and_check([Index, Index, ...]) Lists are sorted and converted to Index. - [Index, Index, ...]: Return ([Index, Index, ...], TYPE) TYPE = 'special' if at least one special type, 'array' otherwise. Parameters ---------- indexes : list of Index or list objects Returns ------- sanitized_indexes : list of Index or list objects type : {'list', 'array', 'special'} r2rhr`ra)rhtyper4r>r remove)r6r7kindsrps r'rjrjJs* 73%$u+3 4E u} u:>KRFGjE&:d1gAG LL F? " 5zA~e+ !!4s B$-B)cXt|}t|tfd|DS)z Determine if all indexes contain the same elements. Parameters ---------- indexes : iterable of Index objects Returns ------- bool True if all indexes contain the same elements, False otherwise. c3@K|]}j|ywrGrc)rHr7firsts r'rJz#all_indexes_same..s4uu||E"4re)iternextrR)r6itrrys @r'rrps( w-C IE 44 44r0cFt|}tj|dS)N)r;)ranger _simple_new)nrngs r'rrs (C  ! !#D 11r0)FrTF) rboolr$rrrr rrfr )r# list[Index]rfr)FFF) r6rrrrrr rrfr )r7r rfr )T)rz bool | Nonerfr )rfr)rintrfr)2 __future__rtextwraptypingrrnumpyr? pandas._libsrr pandas.errorsrpandas.core.dtypes.castr pandas.core.algorithmsr pandas.core.indexes.baser r r rrpandas.core.indexes.categoryrpandas.core.indexes.datetimesrpandas.core.indexes.intervalrpandas.core.indexes.multirpandas.core.indexes.periodrpandas.core.indexes.rangerpandas.core.indexes.timedeltasrpandas._typingrdedent _sort_msg__all__rr/r"rrrjrrr0r'rs" ,4,:760209# HOO   2 UU U  U  U  UD   - -- -  -  -`D}/@# L5$2r0