o 5c(@sbddlmZddlZddlmZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZddl m!Z!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*e+dZ,gdZ- d4d5ddZ.d6d d!Z/   d7d8d#d$Z0d9d&d'Z1d:d;d)d*Z2d+d,Z3d) annotationsN)cast)NaTlib)InvalidIndexErrorfind_common_type)is_dtype_equal) safe_sort)Index _new_Index ensure_indexensure_index_from_sequencesget_unanimous_names)CategoricalIndex) DatetimeIndex) IntervalIndex) MultiIndex) Float64Index Int64Index NumericIndex UInt64Index) PeriodIndex) RangeIndex)TimedeltaIndexzSorting 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 rrrrrrrrrrrrr rr rget_objs_combined_axis union_indexesrall_indexes_same default_indexsafe_sort_indexFT intersectboolsortcopyreturnr cs"fdd|D}t||||dS)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 csg|]}|qS) _get_axis).0objaxisr%M/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/core/indexes/api.py hz*get_objs_combined_axis..)r r"r#)_get_combined_index)objsr r*r"r# obs_idxesr%r)r+rNsrr/ list[Index]cCs<t}g}|D]}t||vr|t|||q|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r/idsresr(r%r%r+_get_distinct_objsls  r8indexescCst|}t|dkrtg}n*t|dkr|d}n|r/|d}|ddD]}||}q&n t|dd}t|}|r?t|}|rE|}|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 intersectionrr rr#)r9r r"r#indexotherr%r%r+r.zs"      r.r>cCsz|jr|Szt|}Wn tyY|Swt|tr|Sttj|}t|tr2tj||j d}|St ||j |j d}|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 TypeError isinstancerrnpndarray from_tuplesr@r rArB)r> array_sortedr%r%r+rs       r bool | Nonec stdkr tdtdkr!d}t|trtt|}|St\}dfdd }fdd }|d krd}|d d D}d d |D}t|dt|fvrZtdt|tkr{dtfddDsvdd Dd}nt|dkrddd Dd}ddD] }|j |rdndd}q|S|dkr|} dtfddddDs|| t d} | j krԈ | S|} || S)a) Return the union of indexes. The behavior of sort and names is not consistent. Parameters ---------- indexes : list of Index or list objects sort : bool, default True Whether the result index should come out sorted or not. Returns ------- Index rz#Must have at least 1 Index to unionr:r$r cs,ddttjfdd|Dd|dS)a3 Convert indexes to lists and concatenate them, removing duplicates. The final dtype is inferred. Parameters ---------- inds : list of Index or list objects dtype : dtype to set for the resulting Index Returns ------- Index cSst|tr |}|SN)rEr tolist)ir%r%r+convs z4union_indexes.._unique_indices..convcsg|]}|qSr%r%)r'rMrNr%r+r,z:union_indexes.._unique_indices..r;)rB)r rfast_unique_multiple_list)indsrBr;rOr+_unique_indicess z&union_indexes.._unique_indicescs&ddD}|rt|}|Sd}|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 cSsg|] }t|tr|jqSr%)rEr rB)r'idxr%r%r+r, szCunion_indexes.._find_common_index_dtype..Nr)rRdtypesrB)r9r%r+_find_common_index_dtypes  z/union_indexes.._find_common_index_dtypespecialcSsg|] }t|tr|qSr%)rErr'xr%r%r+r,z!union_indexes..cSsg|] }|jdur|qSrK)tzrXr%r%r+r,rZz0Cannot join tz-naive with tz-aware DatetimeIndexTc3s|] }t|jjVqdSrK)r rBrXfirstr%r+ #sz union_indexes..cSsg|]}|dqS)UTC) tz_convertrXr%r%r+r,'r-FcSsg|] }|jtddqS)F)r#)astypeobjectrXr%r%r+r,2rZNr;arrayc3|]}|VqdSrKequals)r'r?)r>r%r+r^<)r$r ) r<AssertionErrorrElistr sorted_sanitize_and_checkrDallunionrrArename) r9r"resultkindrSrVdtisdti_tzsr?rBrAr%)r]r>r9r"r+rsP             rcCsltdd|D}t|vr$t|dkr dd|D}|tn|dfSt|dks.t|vr2|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'} cSsh|]}t|qSr%)typer'r>r%r%r+ ]rPz&_sanitize_and_check..r:cSs&g|]}t|tstt|n|qSr%)rEr rirXr%r%r+r,asz'_sanitize_and_check..rirWrc)rir<remover )r9kindsr%r%r+rkHs  rkcs&t|}t|tfdd|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. c3rdrKrertr\r%r+r^}rgz#all_indexes_same..)iternextrl)r9itrr%r\r+rns rnintrcCstd|}tj|ddS)Nr)rA)ranger _simple_new)r{rngr%r%r+rs r)FrTF)r r!r"r!r#r!r$r )r/r1r$r1)FFF) r9r1r r!r"r!r#r!r$r )r>r r$r )T)r"rJr$r )r$r!)r{r|r$r)6 __future__rtextwraptypingrnumpyrF pandas._libsrr pandas.errorsrpandas.core.dtypes.castrpandas.core.dtypes.commonr pandas.core.algorithmsr pandas.core.indexes.baser r r rrpandas.core.indexes.categoryrpandas.core.indexes.datetimesrpandas.core.indexes.intervalrpandas.core.indexes.multirpandas.core.indexes.numericrrrrpandas.core.indexes.periodrpandas.core.indexes.rangerpandas.core.indexes.timedeltasrdedent _sort_msg__all__rr8r.rrrkrrr%r%r%r+sD                0 "| &