o Tj(@sXddlmZddlZddlmZmZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$erpddl%m&Z&e'dZ(gdZ)   d5d6ddZ*d7d!d"Z+   d8d9d$d%Z,d:d'd(Z-d;dd3d4Z1dS)?) 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_indexFT intersectboolaxisrsortcopyreturnr cs"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) _get_axis).0objrr#Z/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/core/indexes/api.py fsz*get_objs_combined_axis..)rr r!)_get_combined_index)objsrrr r! obs_idxesr#r'r(rHsrr+ list[Index]cCs<t}g}|D]}t||vr|t|||q|S)z^ Return a list with distinct elements of "objs" (different ids). Preserves order. )setidaddappend)r+idsresr&r#r#r(_get_distinct_objsjs  r4indexescCst|}t|dkrtg}n*t|dkr|d}n|r/|d}|ddD]}||}q&n t|dd}t|}|r?t|}|rE|}|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 )r4lenr intersectionrr rr!)r5rr r!indexotherr#r#r(r*xs"      r*r:cCsz|jr|Szt|}Wn tyY|Swt|tr|Sttj|}t|tr2tj ||j d}|St||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) is_monotonic_increasingr TypeError isinstancer rnpndarrayr from_tuplesr<r=r>)r: array_sortedr#r#r(rs       r bool | Nonec stdkr tdtdkr)d}t|tr's!t|}|Stt|}|St\}dfdd }fdd }|d krd}d d D}d d |D}t|dt|fvr`tdt|tkrodd}nt|dkrddd Dd}ddD] }|j|rdndd}q|S|dkr|} dt fddddDs|| t d} | j krȈ | S|} || 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 unionr6r"r cstdd|Dr?fdd|D}|d}|d|dd}|||d k}t|r7||}r=|}|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)rAr r%indr#r#r( z9union_indexes.._unique_indices..csg|] }|jddqSF)r!)astyperHr>r#r(r)z:union_indexes.._unique_indices..rr6NcSst|tr |}|SrG)rAr tolist)ir#r#r(convs z4union_indexes.._unique_indices..convcsg|]}|qSr#r#)r%rS)rTr#r(r)r7rN) alluniquer1get_indexer_forr8 sort_valuesr rfast_unique_multiple_list)indsr>resultr;diffr7)rTr>r(_unique_indicess  z&union_indexes.._unique_indicescs&ddD}|rt|}|Sd}|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#)rAr r>)r%idxr#r#r(r)szCunion_indexes.._find_common_index_dtype..Nr)r[dtypesr>)r5r#r(_find_common_index_dtype s  z/union_indexes.._find_common_index_dtypespecialcSsg|] }t|tr|qSr#)rArr%xr#r#r(r) rOz!union_indexes..cSsg|] }|jdur|qSrG)tzrcr#r#r(r)!rOz0Cannot join tz-naive with tz-aware DatetimeIndexTFcSsg|] }|jtddqSrL)rMobjectrcr#r#r(r)4rONr7arrayc3|]}|VqdSrGequals)r%r;)r:r#r(rJ>rKz union_indexes..)r"r ) r8AssertionErrorrAlistr sorted_sanitize_and_checkr@unionrVrr=rename) r5r r\kindr^radtisdti_tzsr;r>r=r#)r:r5r r(rsP      "       rcCsltdd|D}t|vr$t|dkr dd|D}|tn|dfSt|dks.t|vr2|dfS|dfS) 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%r:r#r#r( _rUz&_sanitize_and_check..r6cSs&g|]}t|tstt|n|qSr#)rAr rlrcr#r#r(r)csz'_sanitize_and_check..rlrbrg)rlr8remover )r5kindsr#r#r(rnJs  rncs&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. c3rhrGrirufirstr#r(rJrKz#all_indexes_same..)iternextrV)r5itrr#ryr(rps rnintrcCst|}tj|ddS)N)r=)ranger _simple_new)r~rngr#r#r(rsr)FrTF) rrrrr rr!rr"r )r+r-r"r-)FFF) r5r-rrr rr!rr"r )r:r r"r )T)r rFr"r )r"r)r~rr"r)2 __future__rtextwraptypingrrnumpyrB pandas._libsrr pandas.errorsrpandas.core.dtypes.castr pandas.core.algorithmsr pandas.core.indexes.baser r r rrpandas.core.indexes.categoryrpandas.core.indexes.datetimesrpandas.core.indexes.intervalrpandas.core.indexes.multirpandas.core.indexes.periodrpandas.core.indexes.rangerpandas.core.indexes.timedeltasrpandas._typingrdedent _sort_msg__all__rr4r*rrrnrrr#r#r#r(sL              " 0 " &