B W0d#@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).0obj)axisrI/var/www/html/venv/lib/python3.7/site-packages/pandas/core/indexes/api.py _sz*get_objs_combined_axis..)rrr)_get_combined_index)objsrr!rrZ obs_idxesr)r!r"rEsz list[Index])r%rcCs@t}g}x0|D](}t||kr|t|||qW|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r%Zidsresr rrr"_get_distinct_objscs  r+)indexesrrrrcCst|}t|dkrtg}nXt|dkr4|d}nB|rb|d}x4|ddD]}||}qNWnt||d}t|}|ry |}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 rN)r) 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 Sx|dd D]}| |}qW|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-r)rcs(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|}|S)N) isinstancertolist)irrr"convs z4union_indexes.._unique_indices..convcsg|] }|qSrr)rr5)r6rr"r#sz:union_indexes.._unique_indices..)r)rrZfast_unique_multiple_list)Zinds)r)r6r"_unique_indicessz&union_indexes.._unique_indicesspecial union_manyNarrayc3s|]}|VqdS)N)equals)rr2)r1rr" sz union_indexes..)r.AssertionErrorr3listrsorted_sanitize_and_checkhasattrr9unionallr namerename)r,rresultkindr7r2rDr)r1rr"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)type)rr1rrr" sz&_sanitize_and_check..r-cSs&g|]}t|tstt|n|qSr)r3rr>)rxrrr"r#sz'_sanitize_and_check..r>r8r:N)r>r.remover)r,kindsrrr"r@s  r@)rcs&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|]}|VqdS)N)r;)rr1)firstrr"r<!sz#all_indexes_same..)iternextrC)r,Zitrr)rMr"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$rr@rrrrr"sV         0G&