U
|
¬ý°d( ã @ s d dl mZ d dlZd dlmZ d dlZd dlmZm Z d dl
|
m Z d dl m Z d dlmZ d dlmZ d d lmZmZmZmZmZ d d
|
lmZ d d lmZ d d l m Z d d l m Z d dl m!Z! d dl"m#Z# d dl$m%Z% e &d¡Z'ddddddddddd d d d d d!d"d#d$gZ(d>d'd(d'd'dd)d*d Z)d+d+d,d-d.Z*d?d+d'd'd'dd/d0d1Z+ddd2d3d$Z,d@d4dd5d6d Z-d7d8 Z.d'd9d:d"Z/d;dd<d=d#Z0dS )Aé )Ú annotationsN)Úcast)ÚNaTÚlib)ÚAxis)ÚInvalidIndexError©Úfind_common_type)Ú safe_sort)ÚIndexÚ
|
_new_IndexÚ ensure_indexÚensure_index_from_sequencesÚget_unanimous_names)ÚCategoricalIndex)Ú DatetimeIndex)Ú IntervalIndex)Ú
|
MultiIndex)Ú 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 Úget_objs_combined_axisÚ union_indexesr Úall_indexes_sameÚ default_indexÚsafe_sort_indexFTÚboolr )Ú intersectÚaxisÚsortÚcopyÚreturnc s" 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
|
c s g | ]}| ¡qS © )Z _get_axis)Ú.0Úobj©r r" úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\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 D s z list[Index])r) r! c C s< t }g }| D ](}t||kr| t|¡ | |¡ q|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_objsb s r0 )Úindexesr r r r! c C s t | } t| dkr tg }nTt| dkr4| d }n>|r^| d }| dd
D ]}| |¡}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
|
r é NF©r )r0 Úlenr Ú intersectionr r r r )r1 r r r ÚindexÚotherr" r" r& r( p s
|
|
r( )r6 r! c C sx | j r
|
| S z t| }W n tk
|
r* Y nJX t|tr:|S ttj|}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)ÚnameÚdtype) Zis_monotonic_increasingr
|
Ú TypeErrorÚ
|
isinstancer r ÚnpZndarrayr Ú from_tuplesr8 r9 r: )r6 Z array_sortedr" r" r& r s
|
|
z bool | None)r r! c sª t dkrtdt dkrBd }t|tr>tt|}|S t\}ddfdd }fdd }|d
|
kr*d }d d D }d d |D }t |dt |fkr¶tdt |t krÔdd }n&t |dkrúddd D d }dd
D ] }|j|rdndd}q|S |dkr|} d t fdddd
D sn|| t
|
d }
|
|
|
j kr |
|
¡ S |} || S dS )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 unionr2 r ©r! c s, dd t tj fdd| D d|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
|
c S s t | tr| ¡ } | S ©N)r< r Útolist)Úir" r" r& Úconvì s
|
z4union_indexes.<locals>._unique_indices.<locals>.convc s g | ] } |qS r" r" )r# rB ©rC r" r& r' ò s z:union_indexes.<locals>._unique_indices.<locals>.<listcomp>r3 )r: )r r Zfast_unique_multiple_list)Úindsr: r3 rD r& Ú_unique_indicesÜ s
|
þz&union_indexes.<locals>._unique_indicesc s$ dd D }|r t |}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
|
c S s g | ]}t |tr|jqS r" )r< r r: )r# Úidxr" r" r& r' s
|
zCunion_indexes.<locals>._find_common_index_dtype.<locals>.<listcomp>Nr )rE Zdtypesr: )r1 r" r& Ú_find_common_index_dtypeö s
|
|
z/union_indexes.<locals>._find_common_index_dtypeÚspecialc S s g | ]}t |tr|qS r" )r< r ©r# Úxr" r" r& r' s
|
z!union_indexes.<locals>.<listcomp>c S s g | ]}|j d k r|qS r@ )ÚtzrJ r" r" r& r' s
|
z0Cannot join tz-naive with tz-aware DatetimeIndexTFc S s g | ]}|j td dqS )F)r )ZastypeÚobjectrJ r" r" r& r' ! s Nr3 Úarrayc 3 s | ]} |¡V qd S r@ ©Úequals)r# r7 )r6 r" r& Ú <genexpr>+ s z union_indexes.<locals>.<genexpr>) r4 ÚAssertionErrorr< Úlistr ÚsortedÚ_sanitize_and_checkr; ÚunionÚallr r9 Úrename) r1 r ÚresultÚkindrF rH ZdtisZdti_tzsr7 r: r9 r" )r6 r1 r r& r  sJ
|
|
|
|
|
|
c C sp t dd | D }t |krHt|dkr@dd | D } | t ¡ 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# r6 r" r" r& Ú <setcomp>L s z&_sanitize_and_check.<locals>.<setcomp>r2 c S s& g | ] }t |ts tt|n|qS r" )r< r rS rJ r" r" r& r' P s z'_sanitize_and_check.<locals>.<listcomp>rS rI rN N)rS r4 Úremover )r1 Úkindsr" r" r& rU 7 s ÿ rU r? c s& t | }t| t fdd|D S )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.
|
c 3 s | ]} |¡V qd S r@ rO r\ ©Úfirstr" r& rQ l s z#all_indexes_same.<locals>.<genexpr>)ÚiterÚnextrW )r1 Úitrr" r` r& r ] s Úint)Únr! c C s t d| }tj|d dS )Nr )r9 )Úranger Z _simple_new)rf Úrngr" r" r& r o s
|
)Fr TF)FFF)T)1Ú
|
__future__r ÚtextwrapÚtypingr Únumpyr= Z pandas._libsr r Zpandas._typingr Z pandas.errorsr Zpandas.core.dtypes.castr Zpandas.core.algorithmsr
|
Zpandas.core.indexes.baser r 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.periodr Zpandas.core.indexes.ranger Z pandas.core.indexes.timedeltasr ÚdedentZ _sort_msgÚ__all__r r0 r( r r rU r r r" r" r" r& Ú<module> sl ÿ í ÿ ü0"u&
|
