2i^FdZddlmZddlZddlmZmZddlmZm Z m Z m Z m Z m Z ddlZddlmZddlZddlZddlmZmZmZmZmZmZddlZddlmZdd lmZdd l m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*dd l+m,Z,erddl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5dZ6dZ7dAdZ8dZ9dZ:dBdZ;dBdZdCdZ?e dD dEdZ@edDdFdZ@dGdFdZ@ dG dHdZAdZBdIdZCdBd ZDdBd!ZEdJd"ZFdKd#ZGd$ZHd%ZId&ZJedLd'ZKe dMd(ZKdGdNd)ZKed*ZLe dOd+ZMe dPd,ZM dQd-ZMd.ZN dRd/ZOejdSdTd0ZQdUd1ZRiejd2ejd3ejd4ejd5ejd6ejd2ejd2ejd7ejd7ejd8ejd8ejd9ejd9ejd:ejd:ejd;ejd;ejd3ejd3ejd4ejd4ejdZjdWd?Zkd@Zly)Xzh Misc tools for implementing data structures Note: pandas.core.common is *not* part of the public API. ) annotationsN)abc defaultdict)Callable Collection GeneratorHashableIterableSequence)partial) TYPE_CHECKINGAny ConcatenateTypeVarcastoverload)lib)'construct_1d_object_array_from_listlike) is_bool_dtype is_integer)ABCExtensionArrayABCIndex ABCMultiIndexABCNumpyExtensionArray ABCSeries)iterable_not_string) Expression) AnyArrayLike ArrayLikeNpDtypeP RandomStateT)Indexc#dK|D]%}t|rt|Ed{"|'y7 w)a Flatten an arbitrarily nested sequence. Parameters ---------- line : sequence The non string sequence to flatten Notes ----- This doesn't consider strings sequences. Returns ------- flattened : generator N)rflatten)lineelements I/var/www/html/acx/venv/lib/python3.12/site-packages/pandas/core/common.pyr&r&Ds5" w 'w' ' 'M  's 0. 0c|dj}|ddD]} |j|k7rd}|S|S#t$rd}Y|SwxYw)Nr)name ValueError)objsr,objs r)consensus_name_attrr0\sn 7<znot_none..s 3C3?C 3srKargss r)not_nonerRs 44 33rBc&td|DS)z? Returns a boolean indicating if any argument is None. c3$K|]}|du ywrJrKrLs r)rOzany_none..+ssd{+anyrPs r)any_nonerY +d+ ++rBc&td|DS)zA Returns a boolean indicating if all arguments are None. c3$K|]}|du ywrJrKrLs r)rOzall_none..rUrVallrPs r)all_noner_rZrBc&td|DS)zC Returns a boolean indicating if any argument is not None. c3$K|]}|du ywrJrKrLs r)rOzany_not_none../3s$/rVrWrPs r) any_not_nonerc /$/ //rBc&td|DS)zE Returns a boolean indicating if all arguments are not None. c3$K|]}|du ywrJrKrLs r)rOzall_not_none..rbrVr]rPs r) all_not_nonergrdrBc&td|DS)z; Returns the count of arguments that are not None. c3$K|]}|du ywrJrK)rMxs r)rOz!count_not_none..s+q}+rV)sumrPs r)count_not_nonerlrZrBcyrJrKvaluesr6s r)asarray_tuplesaferpsrBcyrJrKrns r)rprpsSVrBct|ttfst|ds t|}n8t|tr |j St|t r |j St|tr#|tjtfvr t|S tj||}t|jjt rtj|t}|j"dk(r#|Dcgc] }t|}}t|}|S#t$rt|cYSwxYwcc}w)N __array__r6)r3r:tuplehasattrr_valuesrr4r7objectrr8r- issubclassr6r<strndim)ror6resultrjs r)rprps ve} -1Mf FH %~~ FI &~~&$Ebjj&-A$A6v>>?F%0&,,##S)F&1 {{a$*+q%(++8@ M ? 7v>> ?,sD%D?%D<;D<ct|ttfr|g}t|ttj fs t|}t||}|S#t $r|g}YwxYw)z Transform label or iterable of labels to array, for use in Index. Parameters ---------- dtype : dtype If specified, use as dtype of the resulting array, otherwise infer. Returns ------- array rt)r3r{rvr:r4r5 TypeErrorrp)labelsr6rlabelss r)index_labels_to_arrayrse&3,' ftRZZ0 1 &\F e4G N  XF s A A$#A$c<|t|ttfs|gS|SrJ)r3rvr:r/s r)maybe_make_listr+s  z#t}=u JrBct|tjr%t|tjs t |St t |}|S)zB If obj is Iterable but not list-like, consume into list. )r3rr Sizedr:rrrs r)maybe_iterable_to_listr1s:#s||$ZSYY-GCy z3 C JrBct|txr.|jduxr|jduxr|jduS)z We have a null slice. Nr3slicestartstopsteprs r) is_null_slicer;sH 3  II   HH   HH  rBct|txr9|jduxr)|jduxr|j|jk(S)z> We have an empty slice, e.g. no values are selected. N)r3rrrrs r)is_empty_slicerGsL 3 " IIT ! " HHD  " II ! rBc#ZK|D]"}t|txr t| $yw)z; Find non-trivial slices in "line": yields a bool. N)r3rr)r'ks r)is_true_slicesrSs1<E";=+;';;.fs ;!9 HrB)r3rrr)rrs` r)get_rename_functionrs'  6CKK#;<1H&HrBct|ttjtt t fr|St|tjrt|ts t|S|gS)z Convert list-like or scalar input to list-like. List, numpy and pandas array-like inputs are returned unmodified whereas others are converted to list. ) r3r:r4r5rrrrr r{)ros r)convert_to_list_liker&sK&4XyBSTU FCLL )*VS2IF| 8OrBc#K|rt||}t||| ||rt||yy#|rt||wwxYww)a Temporarily set attribute on an object. Parameters ---------- obj : object Object whose attribute will be modified. attr : str Attribute to modify. value : Any Value to temporarily set attribute to. condition : bool, default True Whether to set the attribute. Provided in order to not have to conditionally use this context manager. Yields ------ object : obj with modified attribute. N)getattrsetattr)r/attrvalue condition old_values r) temp_setattrr5sU*C& T5!*  Cy ) 9 Cy ) sA 4A AA czt|t|k7r$tdt|dt|dy)zC Check the length of data matches the length of the index. zLength of values (z") does not match length of index ()N)r;r-)dataindexs r)require_length_matchrTsI 4yCJ D {E |1   rBrkmaxminr^rXmeanprodstdvarmediancumprodcumsumc,tj|S)zH if we define an internal function for this argument, return it ) _cython_tableget)rNs r)get_cython_funcr~s   S !!rBcXt|Dcgc]\}}|d|n|c}}Scc}}w)a If a name is missing then replace it by level_n, where n is the count Parameters ---------- names : list-like list of column names or None values. Returns ------- list list of column names with the None values replaced. level_) enumerate)namesir,s r)fill_missing_namesrs2CLEBR Swq$DLfQCLd 2 SS Ss&cxtjd}|jjD]}||usyy)a Helper function used in detecting chained assignment. If the pandas object (DataFrame/Series) is a local variable in the caller's frame, it should not be a case of chained assignment or method call. For example: def test(): df = pd.DataFrame(...) df["a"] = 1 # not chained assignment Inside ``df.__setitem__``, we call this function to check whether `df` (`self`) is a local variable in `test` frame (the frame calling setitem). If so, we know it is not a case of chained assignment (even when the refcount of `df` is below the threshold due to optimization of local variables). ruTF)sys _getframef_localsro)r/framevs r)is_local_in_caller_framers=& MM! E ^^ " " $ 8 rB)r>rreturnbool)rr)rint).)rozArrayLike | list | tuple | zipr6NpDtype | Noner np.ndarray)ror r6rrrrJ)rznp.ndarray | Iterabler6rrr)r/zIterable[T] | TrzCollection[T] | T)r'z abc.Iterablerzabc.Generator[bool, None, None])r'rrr)rnp.random.Generatorrr)rzHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Nonerznp.random.RandomState)rzRandomState | None) r/rrzCallable[Concatenate[_T, P], T]rQzP.argsrzP.kwargsrr#) r/rrztuple[Callable[..., T], str]rQrrrrr#) r/rrz>Callable[Concatenate[_T, P], T] | tuple[Callable[..., T], str]rQrrrrr#)roz"Hashable | Iterable | AnyArrayLikerzlist | AnyArrayLike)T)rr{rrrzGenerator[None])rr$rNone)rNrrz str | None)rzSequence[Hashable | None]rzlist[Hashable])m__doc__ __future__rbuiltins collectionsrrcollections.abcrrrr r r contextlib functoolsr rrtypingr rrrrrnumpyr4 pandas._libsrpandas.core.dtypes.castrpandas.core.dtypes.commonrrpandas.core.dtypes.genericrrrrrpandas.core.dtypes.inferencerpandas.core.colrpandas._typingrrr r!r"r#pandasr$r&r0rArGrRrYr_rcrgrlrprrrrrrrrrrrrrrrcontextmanagerrrrkrrr^rXnansumrnanmeanrnanprodrnanstdrnanvarr nanmediannanmaxnanminr nancumprodr nancumsumrrrrrKrBr)r s # K=&0 5p,4,,00, DG *3A  V V@<@ !*8:   < "&@ H H  S     F T]   )            &         &* &* H&* &* &*  &*R I  .   **<   LL% LL% LL%FFE  FFE  FFE IIuGGVJJGGVJJFFEIIuFFEIIu IIx!"LL(#$FFEIIuFFEIIuJJ MM9IIxLL(3 :"T"rB