a Of(@s~ddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$erddl%m&Z&e'dZ(gdZ)d3ddddddddZ*dddddZ+d4ddddddd d!Z,ddd"d#d$Z-d5d%dd&d'd(Z.d)d*Z/dd+d,d-Z0d.d/d0d1d2Z1dS)6) annotationsN) TYPE_CHECKINGcast)NaTlib)InvalidIndexErrorfind_common_type) safe_sort)Index _new_Index ensure_indexensure_index_from_sequencesget_unanimous_names)CategoricalIndex) DatetimeIndex) IntervalIndex) MultiIndex) PeriodIndex) RangeIndex)TimedeltaIndex)AxiszSorting 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 rrrrrrrrr rr rget_objs_combined_axis union_indexesrall_indexes_same default_indexsafe_sort_indexFTboolrr ) intersectaxissortcopyreturncs"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).0objrr#P/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/indexes/api.py fz*get_objs_combined_axis..)rr r!)_get_combined_index)objsrrr r!Z obs_idxesr#r&r'rHsrz list[Index])r+r"cCs<t}g}|D](}t||vr|t|||q|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r+Zidsresr%r#r#r'_get_distinct_objsjs  r1)indexesrr 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 )r1lenr intersectionrr rr!)r2rr r!indexotherr#r#r'r*xs      r*)r7r"cCsv|jr |Sz t|}Wnty(YnJ0t|tr8|Sttj|}t|tr`tj ||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 isinstancer rnpZndarrayr from_tuplesr9r:r;)r7Z array_sortedr#r#r'rs     rz bool | None)r r"c stdkrtdtdkrPd}t|trLs@t|}n tt|}|St\}ddfdd }fdd }|d kr:d}d d D}d d |D}t|dt|fvrtdt|tkrdd}n(t|dkr ddd Dd}ddD]}|j|r*dndd}q|S|dkr|} dt fddddDs~|| 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 unionr3r r"cstdd|Dr~fdd|D}|d}|d|dd}|||d k}t|rn||}rz|}|Sd d ttjfd d|Dd dS)z Concatenate indices and remove duplicates. Parameters ---------- inds : list of Index or list objects dtype : dtype to set for the resulting Index Returns ------- Index css|]}t|tVqdSN)r=r r$indr#r#r' r)z9union_indexes.._unique_indices..csg|]}|jddqSF)r!)astyperBr;r#r'r(r)z:union_indexes.._unique_indices..rr3NcSst|tr|}|SrA)r=r tolist)ir#r#r'convs z4union_indexes.._unique_indices..convcsg|] }|qSr#r#)r$rK)rLr#r'r(r)r4rG) alluniquer/Zget_indexer_forr5Z sort_valuesr rZfast_unique_multiple_list)indsr;resultr8diffr4)rLr;r'_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(r)zCunion_indexes.._find_common_index_dtype..Nr)rOZdtypesr;)r2r#r'_find_common_index_dtype s  z/union_indexes.._find_common_index_dtypespecialcSsg|]}t|tr|qSr#)r=rr$xr#r#r'r( r)z!union_indexes..cSsg|]}|jdur|qSrA)tzrVr#r#r'r(!r)z0Cannot join tz-naive with tz-aware DatetimeIndexTFcSsg|]}|jtddqSrE)rFobjectrVr#r#r'r(4r)Nr4arrayc3s|]}|VqdSrAequals)r$r8)r7r#r'rD>r)z union_indexes..) r5AssertionErrorr=listr sorted_sanitize_and_checkr<unionrMrr:rename) r2r rPkindrRrTZdtisZdti_tzsr8r;r:r#)r7r2r r'rsN      "         rcCsptdd|D}t|vrHt|dkr@dd|D}|tn|dfSt|dks\t|vrd|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$r7r#r#r' _r)z&_sanitize_and_check..r3cSs&g|]}t|tstt|n|qSr#)r=r r^rVr#r#r'r(csz'_sanitize_and_check..r^rUrZN)r^r5remover )r2kindsr#r#r'r`Js  r`r@cs&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|]}|VqdSrAr[refirstr#r'rDr)z#all_indexes_same..)iternextrM)r2itrr#rir'rps rintr)nr"cCst|}tj|ddS)N)r:)rangerZ _simple_new)rorngr#r#r'rsr)FrTF)FFF)T)2 __future__rtextwraptypingrrnumpyr>Z pandas._libsrrZ pandas.errorsrZpandas.core.dtypes.castr Zpandas.core.algorithmsr Zpandas.core.indexes.baser r r rrZpandas.core.indexes.categoryrZpandas.core.indexes.datetimesrZpandas.core.indexes.intervalrZpandas.core.indexes.multirZpandas.core.indexes.periodrZpandas.core.indexes.rangerZpandas.core.indexes.timedeltasrZpandas._typingrdedentZ _sort_msg__all__rr1r*rrr`rrr#r#r#r'sJ             "0"&