3
���h� � @ sr d dl Z d dlmZmZ d dlmZmZ d dlmZ d dl j
j Z d dl
mZmZmZmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZm Z m Z m Z d d l m Z d d l!m"Z" d d
l#m$Z$ e j%d�Z&dddddddddddddd d d d d d!d"d#gZ'd4e(e(e(ed&�d'd �Z)ee ee d(�d)d*�Z*d5ee e(e(e(ed+�d,d-�Z+d6ed.�d/d!�Z,d0d1� Z-d2d"� Z.d3d#� Z/dS )7� N)�List�Set)�NaT�lib)�InvalidIndexError)�Index�
_new_Index� ensure_index�ensure_index_from_sequences)�CategoricalIndex)�
DatetimeIndex)�
IntervalIndex)�
MultiIndex)� Float64Index�
Int64Index� NumericIndex� UInt64Index)� PeriodIndex)�
RangeIndex)�TimedeltaIndexz�Sorting 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 r r r r r r
r r r r r r r r r r
�get_objs_combined_axis�
union_indexes�get_consensus_names�all_indexes_sameFT)� intersect�sort�copy�returnc s" � fdd�| D �}t ||||d�S )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
c s g | ]}|j � ��qS � )Z _get_axis)�.0�obj)�axisr �9/tmp/pip-build-5_djhm0z/pandas/pandas/core/indexes/api.py�
<listcomp>[ s z*get_objs_combined_axis.<locals>.<listcomp>)r r r )�_get_combined_index)�objsr r! r r Z obs_idxesr )r! r" r A s )r% r c C s@ t � }g }x0| D ](}t|�|kr|jt|�� |j|� qW |S )z^
Return a list with distinct elements of "objs" (different ids).
Preserves order.
)�set�id�add�append)r% �ids�resr r r r" �_get_distinct_objs_ s
r, )�indexesr r r r c C s� t | �} t| �dkr tg �}nXt| �dkr4| d }nB|rb| d }x4| dd� D ]}|j|�}qNW nt| |d�}t|�}|r�y |j� }W n tk
r� Y nX |r�|j� }|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
r � N)r ) r, �lenr � intersectionr r Z sort_values� TypeErrorr )r- r r r �index�otherr r r" r$ m s&
r$ )r c s t | �dkrtd��t | �dkrB| d }t|t�r>tt|��}|S t| �\} }td�� fdd� }|dkr�| d }t|d�r�|j| dd � �S x | dd � D ]}|j |�}q�W |S nj|d
k�r| d }x&| dd � D ]}|j
|�s�|| �S q�W t | �d }||j k�r|j
|d �}|S || �S d 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
r z#Must have at least 1 Index to unionr. )r c s( dd� � t tj� fdd�| D ��d��S )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
c S s t | t�r| j� } | S )N)�
isinstancer �tolist)�ir r r" �conv� s
z4union_indexes.<locals>._unique_indices.<locals>.convc s g | ] }� |��qS r r )r r6 )r7 r r" r# � s z:union_indexes.<locals>._unique_indices.<locals>.<listcomp>)r )r r Zfast_unique_multiple_list)Zinds)r )r7 r" �_unique_indices� s z&union_indexes.<locals>._unique_indices�special�
union_manyN�array)�name)r/ �AssertionErrorr4 �listr �sorted�_sanitize_and_check�hasattrr: �union�equalsr r< Z
_shallow_copy)r- r �result�kindr8 r3 r2 r< r )r r" r � s4
c C sp t dd� | D ��}t |krHt|�dkr@dd� | D �} |jt � n| dfS t|�dks\t|krd| dfS | dfS d 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'}
c S s h | ] }t |��qS r )�type)r r2 r r r" � <setcomp>� s z&_sanitize_and_check.<locals>.<setcomp>r. c S s& g | ] }t |t�s tt|��n|�qS r )r4 r r> )r �xr r r" r# s z'_sanitize_and_check.<locals>.<listcomp>r> r9 r; N)r> r/ �remover )r- Zkindsr r r" r@ � s r@ c C s: dd� | D �}t |�dkr*tt|�d �S dg| d j S )a%
Give a consensus 'names' to indexes.
If there's exactly one non-empty 'names', return this,
otherwise, return empty.
Parameters
----------
indexes : list of Index objects
Returns
-------
list
A list representing the consensus 'names' found.
c S s" h | ]}t j|j� rt|j��qS r )�comZ any_not_none�names�tuple)r r6 r r r" rG ! s z&get_consensus_names.<locals>.<setcomp>r. r N)r/ r> Znlevels)r- Zconsensus_namesr r r" r s c C s0 t | �}t|�}x|D ]}|j|�sdS qW 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.
FT)�iter�nextrC )r- Zitr�firstr2 r r r" r ' s
)Fr TF)FFF)T)0�textwrapZtypingr r Z pandas._libsr r Z
pandas.errorsr Zpandas.core.common�core�commonrJ Zpandas.core.indexes.baser r r r
Z pandas.core.indexes.categoryr Z pandas.core.indexes.datetimesr Z pandas.core.indexes.intervalr
Zpandas.core.indexes.multir Zpandas.core.indexes.numericr r r r Zpandas.core.indexes.periodr Zpandas.core.indexes.ranger Z pandas.core.indexes.timedeltasr �dedentZ _sort_msg�__all__�boolr r, r$ r r@ r r r r r r" �<module> s^ /H& |