*j/%RddlmZddlmZmZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$erddl%m&Z&gdZ' d ddZ(ddZ) d ddZ*ddZ+d d!dZ,dZ-d"dZ.d#dZ/y)$) annotations) TYPE_CHECKINGcastN)NaTlib)InvalidIndexError)find_common_type) safe_sort)Index _new_Index ensure_indexensure_index_from_sequencesget_unanimous_namesmaybe_sequence_to_range)CategoricalIndex) DatetimeIndex) IntervalIndex) MultiIndex) PeriodIndex) RangeIndex)TimedeltaIndex)Axis)rrr rrrrrrrr all_indexes_same default_indexr rget_objs_combined_axisrrsafe_sort_index union_indexescd|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. NoDefault use for deprecation in GH#57335. Returns ------- Index ) intersectsort) _get_axis_get_combined_index)objsraxisr obj obs_idxess 9C:\FundamentalWorkbench\vendor\pandas\core\indexes\api.pyrr?s58155t$I5 yID II6s-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/_sH EC C c7#  GGBsG  JJsO JcTt|}t|dk(r td}nct|dk(r|d}nO|r!|d}|ddD]}|j|}n,t ||t j ur|nd}t|}|r|t j ur t|}|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. NoDefault used for deprecation of GH#57335 Returns ------- Index rNFr ) r/lenr intersectionrr no_defaultr r)indexesrr indexothers r'r"r"ms0!)G 7|q$Q' W    QR[E&&u-E!gDCNN4JDPUVU# CNN*& 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)r8 array_sorteds r'rrs $$  L '  lE * BJJ 5 eZ (**z union_indexes..s'XPWZU%;SZZ\DPWs,.)r=zkind=z& must be 'special', 'array' or 'list'.)r4AssertionErrorr?rKrr6r sorted_sanitize_and_checkrtzrCastypeobjectunionrr r=uniquer,get_indexer_for sort_valuesrr<renamefast_unique_multiple_list_genbool ValueError)r7r resultkindnum_dtis num_dti_tzsrPxr9r=indindsr8diffr<dtypes all_listss r'rrs" 7|qBCC 7|q fd #43>>1v vf~. '0MGT y C#}-A 66%1$K  q(m + NO O s7| #s~~%QZF \D>EEWqxxUx3WGEQZFQR[E\\%dd\FF! ($7%C7Ccii7%CDE=DEWcCJJu5J1WDEGNN$EGNN48,E..u5;|Dcgc](}t|tst t|n|*}}|thz}n|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'} r2rKrFrH)typerKr4r?r )r7r8kindsrds r'rTrTs*'. .gUT%[gE . u} u:>KRKRajE&:d1gA7  dVOEF? " 5zA~e+ !! /s B-BcXt|}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|ywrM)equals)rOr8firsts r'rQz#all_indexes_same..Qs4uu||E"s)iternextall)r7itrrps @r'rrBs( w-C IE 44 44r0cFt|}tj|dS)N)r<)ranger _simple_new)nrngs r'rrTs (C  ! !#D 11r0)FrT)rr^r$rr bool | lib.NoDefaultreturnr )r# list[Index]r{r|)FF)r7r|rr^r rzr{r )r8r r{r )T)r rzr{r )r{r^)rxintr{r)0 __future__rtypingrrnumpyr@ pandas._libsrr pandas.errorsrpandas.core.dtypes.castr pandas.core.algorithmsr pandas.core.indexes.baser r r rrrpandas.core.indexes.categoryrpandas.core.indexes.datetimesrpandas.core.indexes.intervalrpandas.core.indexes.multirpandas.core.indexes.periodrpandas.core.indexes.rangerpandas.core.indexes.timedeltasrpandas._typingr__all__rr/r"rrrTrrr0r'rs" ,4,:760209# 4!% JJ J  J  J@  !&' '' ' 'TD`KF# L5$2r0