E g(`ddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$erddl%m&Z&ej'dZ(gdZ) d-d.dZ*d/d Z+ d0d1d"Z,d2d$Z-d3d4d&Z.d'Z/d5d(Z0d6d,Z1dS)7) annotationsN) TYPE_CHECKINGcast)NaTlib)InvalidIndexErrorfind_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 rrrget_objs_combined_axis union_indexesrall_indexes_same default_indexsafe_sort_indexFT intersectboolaxisrsortcopyreturnr cFfd|D}t||||S)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 c:g|]}|S) _get_axis).0objr s L/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/core/indexes/api.py z*get_objs_combined_axis..fs%555t$$555)rr!r")_get_combined_index)objsrr r!r" obs_idxess ` r*rrHs5<6555555I yIDt T T TTr,r. list[Index]ct}g}|D]J}t||vr7|t|||K|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r.idsresr)s r*_get_distinct_objsr8js_ EEC C c77#   GGBsGG    JJsOOO Jr,indexesct|}t|dkrtg}nit|dkr |d}nM|r+|d}|ddD]}||}n t |d}t |}|rt |}|r|}|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!)r8lenr intersectionrrrr")r9rr!r"indexothers r*r-r-xs4!))G 7||qb  W    $ QRR[ . .E&&u--EE .gE222U## '&&   Lr,r?cV|jr|S t|}t|tr|St t j|}t|trtj||j }n-t||j |j }n#t$rYnwxYw|S)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_tuplesrBrCrD TypeError)r? array_sorteds r*rrs $  L ''  lE * *  BJ 55 eZ ( ( L*z9union_indexes.._unique_indices..s,66#z#u%%666666r,c>g|]}|dSF)r")astype)r(rQrDs r*r+z:union_indexes.._unique_indices..s)BBBcCJJu5J11BBBr,rr;NcXt|tr|}|SrP)rFr tolist)is r*convz4union_indexes.._unique_indices..convs&!U## HHJJHr,c&g|] }|Sr&r&)r(rZr[s r*r+z:union_indexes.._unique_indices..s!*A*A*Aq4477*A*A*Ar,r<)rD) alluniquer5get_indexer_forr= sort_valuesr rfast_unique_multiple_list)indsrDresultr@diffr[r!s ` @r*_unique_indicesz&union_indexes.._unique_indicess$ 66666 6 6 BBBBTBBBD!W^^%%FGNN48,,E//66"<=D4yy 6t{{}}55 .++--M      )*A*A*A*AD*A*A*A M M M    r,cHdD}|rt|}nd}|S)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 cFg|]}t|t|jSr&)rFr rD)r(idxs r*r+zCunion_indexes.._find_common_index_dtype..s)III*S%2H2HI#)IIIr,Nr )rbdtypesrDr9s r*_find_common_index_dtypez/union_indexes.._find_common_index_dtype s:JIwIII  $V,,EEE r,specialc<g|]}t|t|Sr&)rFrr(xs r*r+z!union_indexes.. s'CCCajM&B&BCCCCr,c g|] }|j | SrP)tzrms r*r+z!union_indexes..!s777ad&61&6&6&6r,z0Cannot join tz-naive with tz-aware DatetimeIndexTFcFg|]}|tdSrT)rUobjectrms r*r+z!union_indexes..4s(EEEqxxUx33EEEr,Nr<arrayc3BK|]}|VdSrPequals)r(r@r?s r*rRz union_indexes..>s/@@55<<&&@@@@@@r,)r#r ) r=AssertionErrorrFlistr sorted_sanitize_and_checkrJunionr]rrCrename) r9r!rckindrerjdtisdti_tzsr@rDrCr?s `` @r*rrs  7||qBCCC 7||q fd # # / /vvf~~.. '00MGT       D( yCC7CCC77d777 w<<3t99~ - - NOO O t99G $ $DQZFF YY]]DFEWEEEGQZFQRR[ G GE\\%d.Edd\FFFF ((11 @@@@GABBK@@@@@ 4#OGU33E"G,Q/ 5:  LL&&E ((11w...r,c td|D}t|vr>t|dkr'd|D}|tn|dfSt|dks t|vr|dfS|dfS)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'} c,h|]}t|Sr&)type)r(r?s r* z&_sanitize_and_check.._s333%$u++333r,r;ctg|]5}t|tstt|n|6Sr&)rFr rxrms r*r+z'_sanitize_and_check..csDFGjE&:&:Ad1ggr,rxrkrs)rxr=remover )r9kindss r*rzrzJs* 337333 4 4E u}} u::>>KRG LL    F? " 5zzA~~e++ !!r,cvt|}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. c3BK|]}|VdSrPru)r(r?firsts r*rRz#all_indexes_same..s/44uu||E""444444r,)iternextr])r9itrrs @r*rrps> w--C IIE 4444444 4 44r,nintrcLt|}tj|dS)N)rC)ranger _simple_new)rrngs r*rrs# ((C  !#D 1 1 11r,)FrTF) rrr rr!rr"rr#r )r.r0r#r0)FFF) r9r0rrr!rr"rr#r )r?r r#r )T)r!rLr#r )r#r)rrr#r)2 __future__rtextwraptypingrrnumpyrG pandas._libsrr pandas.errorsrpandas.core.dtypes.castr pandas.core.algorithmsr pandas.core.indexes.baser r rrrpandas.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__rr8r-rrrzrrr&r,r*rs"""""" ,+++++444444,,,,,,:99999777777666666000000222222000000999999$###### HO      2 UUUUUD      -----`D}/}/}/}/}/@# # # L5555$222222r,