U c(@sddlmZddlZddlmZddlZddlmZm 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,dddddddddddddd d!d"d#d$d%d&d'd(d)gZ-dBd,d,d,dd-d.d$Z.d/d/d0d1d2Z/dCd/d,d,d,dd3d4d5Z0ddd6d7d)Z1dDd8dd9d:d%Z2d;d<Z3d,d=d>d'Z4d?dd@dAd(Z5dS)E) 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_indexFTbool) intersectsortcopyreturncs"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).0objaxisr%;/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/indexes/api.py hsz*get_objs_combined_axis..)r!r"r#)_get_combined_index)objsr!r)r"r#Z obs_idxesr%r(r*rNsz 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_objsls  r4)indexesr!r"r#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")r4lenr intersectionrr rr#)r5r!r"r#indexotherr%r%r*r,zs      r,)r:r$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 isinstancerrnpZndarray from_tuplesr<r r=r>)r:Z array_sortedr%r%r*rs    z bool | None)r"r$c stdkrtdtdkrBd}t|tr>tt|}|St\}ddfdd }fdd }|d krTd}|d d D}d d |D}t|dt|fkrtdt|tkrdtfddDsdd Dd}n(t|dkr$ddd Dd}ddD]}|j |rDdndd}q0|S|dkr|} dtfddddDs|| 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 unionr6r 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&rFrGr%r*r+sz:union_indexes.._unique_indices..r7)r>)r rZfast_unique_multiple_list)indsr>r7rHr*_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)rIZdtypesr>)r5r%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|qSrD)tzrNr%r%r*r+s z0Cannot join tz-naive with tz-aware DatetimeIndexTc3s|]}t|jjVqdSrD)r r>rNfirstr%r* #sz union_indexes..cSsg|]}|dqS)UTC)Z tz_convertrNr%r%r*r+'sFcSsg|]}|jtddqS)F)r#)ZastypeobjectrNr%r%r*r+2sNr7arrayc3s|]}|VqdSrDequals)r&r;)r:r%r*rS<s) r8AssertionErrorr@listr sorted_sanitize_and_checkr?allunionrr=rename) r5r"resultkindrJrLZdtisZdti_tzsr;r>r=r%)rRr:r5r"r*rsP              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&r:r%r%r* ]sz&_sanitize_and_check..r6cSs&g|]}t|tstt|n|qSr%)r@r rZrNr%r%r*r+asz'_sanitize_and_check..rZrMrVN)rZr8remover )r5kindsr%r%r*r\Hs  r\rCcs&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|]}|VqdSrDrWrcrQr%r*rS}sz#all_indexes_same..)iternextr])r5itrr%rQr*rns int)nr$cCstd|}tj|ddS)Nr)r=)rangerZ _simple_new)rkrngr%r%r*rs )FrTF)FFF)T)6 __future__rtextwraptypingrZnumpyrAZ pandas._libsrrZ pandas.errorsrZpandas.core.dtypes.castrZpandas.core.dtypes.commonr Zpandas.core.algorithmsr Zpandas.core.indexes.baser r r rrZpandas.core.indexes.categoryrZpandas.core.indexes.datetimesrZpandas.core.indexes.intervalrZpandas.core.indexes.multirZpandas.core.indexes.numericrrrrZpandas.core.indexes.periodrZpandas.core.indexes.rangerZpandas.core.indexes.timedeltasrdedentZ _sort_msg__all__rr4r,rrr\rrr%r%r%r*sv              0"|&