U MZf&@sddlmZddlZddlmZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%e&dZ'ddddddddddddddd d!d"d#d$gZ(d>d'd(d'd'dd)d*dZ)d+d+d,d-d.Z*d?d+d'd'd'dd/d0d1Z+ddd2d3d$Z,d@d4dd5d6d Z-d7d8Z.d'd9d:d"Z/d;dd<d=d#Z0dS)A) annotationsN)cast)NaTlib)Axis)InvalidIndexErrorfind_common_type) safe_sort)Index _new_Index ensure_indexensure_index_from_sequencesget_unanimous_names)CategoricalIndex) DatetimeIndex) IntervalIndex) MultiIndex) 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 rrrrrrrrr rr rget_objs_combined_axis union_indexesrall_indexes_same default_indexsafe_sort_indexFTboolr) intersectaxissortcopyreturncs"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)Z _get_axis).0objrr";/tmp/pip-unpacked-wheel-nbcvw55c/pandas/core/indexes/api.py ^sz*get_objs_combined_axis..)rrr )_get_combined_index)objsrrrr Z obs_idxesr"r%r&rDsz list[Index])r)r!cCs<t}g}|D](}t||kr|t|||q|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r)idsresr$r"r"r&_get_distinct_objsbs  r0)indexesrrr r!cCst|}t|dkrtg}nTt|dkr4|d}n>|r^|d}|ddD]}||}qLnt|dd}t|}|r~t|}|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)r0lenr intersectionrr rr )r1rrr indexotherr"r"r&r(ps      r()r6r!cCsx|jr |Sz t|}Wntk r*YnJXt|tr:|Sttj|}t|trbtj ||j d}nt||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) Zis_monotonic_increasingr TypeError isinstancer rnpZndarrayr from_tuplesr8r9r:)r6Z array_sortedr"r"r&rs    z bool | None)rr!c stdkrtdtdkrBd}t|tr>tt|}|St\}ddfdd }fdd }|d kr*d}d d D}d d |D}t|dt|fkrtdt|tkrdd}n&t|dkrddd Dd}ddD]}|j|rdndd}q|S|dkr|} dt fddddDsn|| t d} | j kr | S|} || SdS)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 unionr2r 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)r<r tolist)ir"r"r&convs z4union_indexes.._unique_indices..convcsg|] }|qSr"r")r#rBrCr"r&r'sz:union_indexes.._unique_indices..r3)r:)r rZfast_unique_multiple_list)indsr:r3rDr&_unique_indicess z&union_indexes.._unique_indicescs$ddD}|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 cSsg|]}t|tr|jqSr")r<r r:)r#idxr"r"r&r's zCunion_indexes.._find_common_index_dtype..Nr)rEZdtypesr:)r1r"r&_find_common_index_dtypes  z/union_indexes.._find_common_index_dtypespecialcSsg|]}t|tr|qSr")r<rr#xr"r"r&r' s z!union_indexes..cSsg|]}|jdk r|qSr@)tzrJr"r"r&r's z0Cannot join tz-naive with tz-aware DatetimeIndexTFcSsg|]}|jtddqS)F)r )ZastypeobjectrJr"r"r&r'!sNr3arrayc3s|]}|VqdSr@equals)r#r7)r6r"r& +sz union_indexes..) r4AssertionErrorr<listr sorted_sanitize_and_checkr;unionallrr9rename) r1rresultkindrFrHZdtisZdti_tzsr7r:r9r")r6r1rr&rsJ               cCsptdd|D}t|krHt|dkr@dd|D}|tn|dfSt|dks\t|krd|dfS|dfSd S) 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#r6r"r"r& Lsz&_sanitize_and_check..r2cSs&g|]}t|tstt|n|qSr")r<r rSrJr"r"r&r'Psz'_sanitize_and_check..rSrIrNN)rSr4remover )r1kindsr"r"r&rU7s  rUr?cs&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. c3s|]}|VqdSr@rOr\firstr"r&rQlsz#all_indexes_same..)iternextrW)r1itrr"r`r&r]s int)nr!cCstd|}tj|ddS)Nr)r9)rangerZ _simple_new)rfrngr"r"r&ros )FrTF)FFF)T)1 __future__rtextwraptypingrZnumpyr=Z pandas._libsrrZpandas._typingrZ pandas.errorsrZpandas.core.dtypes.castr Zpandas.core.algorithmsr Zpandas.core.indexes.baser r r rrZpandas.core.indexes.categoryrZpandas.core.indexes.datetimesrZpandas.core.indexes.intervalrZpandas.core.indexes.multirZpandas.core.indexes.periodrZpandas.core.indexes.rangerZpandas.core.indexes.timedeltasrdedentZ _sort_msg__all__rr0r(rrrUrrr"r"r"r&sl              0"u&