U f/e#@sZddlmZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lmZdd lm Z e!dZ"dddddddddddddddddd d!d"d#gZ#d6d&d&d&dd'd(d Z$d)d)d*d+d,Z%d7d)d&d&d&dd-d.d/Z&d8d&dd0d1d!Z'd2d3Z(d&d4d5d#Z)dS)9) annotationsN)NaTlib)InvalidIndexError)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'. rrrrrr r rrrrrr rrrr get_objs_combined_axis union_indexesr all_indexes_sameFTbool) 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-tiezk1ph/pandas/core/indexes/api.py _sz*get_objs_combined_axis..)rrr)_get_combined_index)objsrr"rrZ obs_idxesrr!r#rEsz list[Index])r&rcCs<t}g}|D](}t||kr|t|||q|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r&idsresr rrr#_get_distinct_objscs  r-)indexesrrrrcCst|}t|dkrtg}nTt|dkr4|d}n>|r^|d}|ddD]}||}qLnt||d}t|}|rz |}Wntk rYnX|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 rNr) r-lenr intersectionrrZ sort_values TypeErrorr)r.rrrindexotherrrr#r%qs&      r%)rrcst|dkrtdt|dkrB|d}t|tr>tt|}|St|\}}ddfdd }|dkr|d}t|d r||dd S|dd D]}| |}q|Snf|d kr |dt fd d |dd Ds||t |d}|j kr |S||Sd 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/rrcs(ddttjfdd|DdS)z Convert indexes to lists and concatenate them, removing duplicates. The final dtype is inferred. Parameters ---------- inds : list of Index or list objects Returns ------- Index cSst|tr|}|SN) isinstancertolist)irrr#convs z4union_indexes.._unique_indices..convcsg|] }|qSrr)rr:r;rr#r$sz:union_indexes.._unique_indices..r0)rrZfast_unique_multiple_list)Zindsr0r<r#_unique_indicessz&union_indexes.._unique_indicesspecial union_manyNarrayc3s|]}|VqdSr7equals)rr5)r4rr# sz union_indexes..)r1AssertionErrorr8listrsorted_sanitize_and_checkhasattrr?unionallr namerename)r.rresultkindr=r5rKr)r4rr#rs2           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)typerr4rrr# sz&_sanitize_and_check..r/cSs&g|]}t|tstt|n|qSr)r8rrE)rxrrr#r$sz'_sanitize_and_check..rEr>r@N)rEr1remover)r.kindsrrr#rGs  rGr6cs&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|]}|VqdSr7rArPfirstrr#rC!sz#all_indexes_same..)iternextrJ)r.itrrrUr#rs )FrTF)FFF)T)* __future__rtextwrapZ pandas._libsrrZ pandas.errorsrZpandas.core.indexes.baserrrr r Zpandas.core.indexes.categoryr Zpandas.core.indexes.datetimesr Zpandas.core.indexes.intervalr Zpandas.core.indexes.multirZpandas.core.indexes.numericrrrrZpandas.core.indexes.periodrZpandas.core.indexes.rangerZpandas.core.indexes.timedeltasrdedentZ _sort_msg__all__rr-r%rrGrrrrr#sd          4G&