eh)DdZddlmZddlZddlmZmZddlmZm Z m Z m Z m Z ddl Z ddlmZddlZddlmZmZmZmZmZddlZddlZddlmZdd lmZdd lmZdd l m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)erddl*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1dZ2dZ3dAdZ4dZ5dZ6dBdZ7dBdZ8dBdZ9dBdZ:dCdZ;e dD dEdZdIdZ?dBd Z@dBd!ZAdJd"ZBdKd#ZCd$ZDd%ZEd&ZFedLd'ZGe dMd(ZGdGdNd)ZG dOd*ZHd+ZI dPd,ZJe j dQ dRd-ZLdSd.ZMejejejejjejejjiZTejd/ejd0ejd1iZUiejd2ejd3ejd4ejd5ejd6ejd2ejd2ejd7ejd7ejd8ejd8ejd9ejd9ejd:ejd:ejd;ejd;ejd3ejd3ejd4ejd4ejdZjd?ZkdUd@Zly)Vzh Misc tools for implementing data structures Note: pandas.core.common is *not* part of the public API. ) annotationsN)abc defaultdict) Collection GeneratorHashableIterableSequence)partial) TYPE_CHECKINGAnyCallablecastoverload)lib)np_version_gte1p24)'construct_1d_object_array_from_listlike) is_bool_dtype is_integer)ABCExtensionArrayABCIndex ABCMultiIndex ABCSeries)iterable_not_string) AnyArrayLike ArrayLikeNpDtype 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 P/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/common.pyr"r">s5" w 'w' ' 'M  's 0. 0c~|dj}|ddD]} |j|k7rd}|S#t$rd}Y&wxYw)Nr)name ValueError)objsr(objs r%consensus_name_attrr,VsX 7<cd|DS)zL Returns a generator consisting of the arguments that are not None. c3&K|] }|| ywN.0args r% znot_none..s 3C3?C 3srGargss r%not_nonerNs 44 33r>c&td|DS)z? Returns a boolean indicating if any argument is None. c3$K|]}|du ywrFrGrHs r%rKzany_none..+ssd{+anyrLs r%any_nonerU +d+ ++r>c&td|DS)zA Returns a boolean indicating if all arguments are None. c3$K|]}|du ywrFrGrHs r%rKzall_none..rQrRallrLs r%all_noner[rVr>c&td|DS)zC Returns a boolean indicating if any argument is not None. c3$K|]}|du ywrFrGrHs r%rKzany_not_none../3s$/rRrSrLs r% any_not_noner_ /$/ //r>c&td|DS)zE Returns a boolean indicating if all arguments are not None. c3$K|]}|du ywrFrGrHs r%rKzall_not_none..r^rRrYrLs r% all_not_nonercr`r>c&td|DS)z; Returns the count of arguments that are not None. c3$K|]}|du ywrFrG)rIxs r%rKz!count_not_none..s+q}+rR)sumrLs r%count_not_nonerhrVr>cyrFrGvaluesr2s r%asarray_tuplesaferlsr>cyrFrGrjs r%rlrlr>c.t|ttfst|ds t|}n8t|tr |j St|t r |j St|tr#|tjtfvr t|S tj5ts$tjdtjtj ||}dddt%j&j(t*rtj |t}|j,dk(r#|Dcgc] }t|}}t|}|S#1swY|xYw#t"$rt|cYSwxYwcc}w)N __array__ignorer2)r/r6tuplehasattrr_valuesrr0r3objectrwarningscatch_warningsr simplefilterVisibleDeprecationWarningr4r) issubclassr2r8strndim)rkr2resultrfs r%rlrls7 ve} -1Mf FH %~~ FI &~~&$Ebjj&-A$A6v>> ?  $ $ & 5%%%h0L0LMZZe4F  5&,,##S)F&1 {{a$*+q%(++8@ M) 5 5 ? 7v>> ?,s1E8/AE,1E8 F,E51E88FFct|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 rr)r/r}rtr6r0r1 TypeErrorrl)labelsr2s r%index_labels_to_arrayr se&3,' ftRZZ0 1 &\FvU 3F M  XF s A A$#A$c<|t|ttfs|gS|SrF)r/rtr6r+s r%maybe_make_listr's  z#t}=u Jr>ct|tjr%t|tjs t |St t |}|S)zB If obj is Iterable but not list-like, consume into list. )r/rr Sizedr6rrrs r%maybe_iterable_to_listr-s:#s||$ZSYY-GCy z3 C Jr>ct|txr.|jduxr|jduxr|jduS)z We have a null slice. Nr/slicestartstopsteprs r% is_null_slicer7sH 3  II   HH   HH  r>ct|txr9|jduxr)|jduxr|j|jk(S)z> We have an empty slice, e.g. no values are selected. N)r/rrrrs r%is_empty_slicerCsL 3 " IIT ! " HHD  " II ! r>cd|Dcgc] }t|txr t| "c}Scc}w)zX Find non-trivial slices in "line": return a list of booleans with same length. )r/rr)r#ks r%is_true_slicesrOs1DH HaJq% 9q)9%9 9 HH Hs%-ct|txr0|jdk(xr|j|k(xr|jduS)z& We have a full length slice. rNr)r+r#s r% is_full_slicerWsG 3  IIN  HH   HH  r>ct|dr t|dSt|trt |j St |rt|jSy)N__name__) rugetattrr/r get_callable_namefunccallabler8rrs r%rrcsNsJsJ''#w **}Cy!!! r>c .t|r ||fi|S|S)z Evaluate possibly callable input using obj and kwargs if it is callable, otherwise return as it is. Parameters ---------- maybe_callable : possibly a callable obj : NDFrame **kwargs )r)maybe_callabler+kwargss r%apply_if_callablerts"c,V,, r>ctj|s5t|trt t|j St |}t|tjstd||tk(r td|S)a Helper function to standardize a supplied mapping. Parameters ---------- into : instance or subclass of collections.abc.Mapping Must be a class, an initialized collections.defaultdict, or an instance of a collections.abc.Mapping subclass. Returns ------- mapping : a collections.abc.Mapping subclass or other constructor a callable object that can accept an iterator to create the desired Mapping. See Also -------- DataFrame.to_dict Series.to_dict zunsupported type: z/to_dict() only accepts initialized defaultdicts) inspectisclassr/rr default_factoryr8r|rMappingr)intos r%standardize_mappingrsp* ??4 dK (;(<(<= =Dz dCKK (,TF344 {IJJ Kr>cyrFrGstates r% random_staterrnr>cyrFrGrs r%rrsr>ct|s4t|tjtjj frtjj |St|tjj r|St|tjjr|S|tjStd)aq Helper function for processing random_state arguments. Parameters ---------- state : int, array-like, BitGenerator, Generator, np.random.RandomState, None. If receives an int, array-like, or BitGenerator, passes to np.random.RandomState() as seed. If receives an np.random RandomState or Generator, just returns that unchanged. If receives `None`, returns np.random. If receives anything else, raises an informative ValueError. Default None. Returns ------- np.random.RandomState or np.random.Generator. If state is None, returns np.random zdrandom_state must be an integer, array-like, a BitGenerator, Generator, a numpy RandomState, or None) rr/r0r1random BitGeneratorrrr)rs r%rrs(%Jurzz299;Q;Q.RSyy$$U++ E29900 1 E299.. / yy +  r>ct|tr&|\}}||vr|d}t||||<||i|S||g|i|S)as Apply a function ``func`` to object ``obj`` either by passing obj as the first argument to the function or, in the case that the func is a tuple, interpret the first element of the tuple as a function and pass the obj to that function as a keyword argument whose key is the value of the second element of the tuple. Parameters ---------- func : callable or tuple of (callable, str) Function to apply to this object or, alternatively, a ``(callable, data_keyword)`` tuple where ``data_keyword`` is a string indicating the keyword of ``callable`` that expects the object. *args : iterable, optional Positional arguments passed into ``func``. **kwargs : dict, optional A dictionary of keyword arguments passed into ``func``. Returns ------- object : the return type of ``func``. z/ is both the pipe target and a keyword argument)r/rtr))r+rrMrtargetmsgs r%pipersd4$ f V HKLCS/ !vT$V$$C)$)&))r>cVfd}ttjtfr|SS)zv Returns a function that will map names/labels, dependent if mapper is a dict, Series or just a function. c|vr|S|SrFrG)rfmappers r%fzget_rename_function..fs ;!9 Hr>)r/rrr)rrs` r%get_rename_functionrs'  6CKK#;<1H&Hr>ct|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. ) r/r6r0r1rrrrr r})rks r%convert_to_list_likersK&4XyBSTU FCLL )*VS2IF| 8Or>c#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)rsetattr)r+attrvalue condition old_values r% temp_setattrrsU.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)r7r))dataindexs r%require_length_matchr8sI 4yCJ D {E |1   r>znp.sumznp.maximum.reduceznp.minimum.reducergmaxminrZrTmeanprodstdvarmediancumprodcumsumc,tj|S)zH if we define an internal function for this argument, return it ) _cython_tablegetrJs r%get_cython_funcrrs   S !!r>c.tj||S)zd if we define a builtin function for this argument, return it, otherwise return the arg )_builtin_tablerrs r%is_builtin_funcrys   c3 ''r>cXt|Dcgc]\}}|d|n|c}}Scc}}w)a, If a name is missing then replace it by level_n, where n is the count .. versionadded:: 1.4.0 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_namesrs2 CLEBR Swq$DLfQCLd 2 SS Ss&)r:r returnbool)rr)rint).)rkzArrayLike | list | tuple | zipr2NpDtype | Noner np.ndarray)rkr r2rrrrF)rznp.ndarray | Iterabler2rrr)r+zIterable[T] | TrzCollection[T] | T)rz list[bool])r#rrr)rnp.random.Generatorrr)rzHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Nonerznp.random.RandomState)rzRandomState | None)rz/Callable[..., T] | tuple[Callable[..., T], str]rr)rkz"Hashable | Iterable | AnyArrayLikerzlist | AnyArrayLike)T)rr}rrrzGenerator[None, None, None])rr rNone)rJrrz str | None)rzSequence[Hashable | None]rzlist[Hashable])m__doc__ __future__rbuiltins collectionsrrcollections.abcrrrr r contextlib functoolsr rtypingr r rrrrxnumpyr0 pandas._libsrpandas.compat.numpyrpandas.core.dtypes.castrpandas.core.dtypes.commonrrpandas.core.dtypes.genericrrrrpandas.core.dtypes.inferencerpandas._typingrrrrrpandasr r"r,r=rCrNrUr[r_rcrhrlrrrrrrrrrrrrrrcontextmanagerrrrgrmaximumreducerminimumr_builtin_table_aliasrZrTnansumrnanmeanrnanprodrnanstdrnanvarr nanmediannanmaxnanminr nancumprodr nancumsumrrrrrGr>r%rs #2K =04n,4,,00, DG *3A    H<@ !*8:   I ""@    S   F"*>"*"*J I  .   -1**&** **@  " LL"&& LL"**## LL"**## LL( LL% LL%  LL% LL% LL%FFE  FFE  FFE IIuGGVJJGGVJJFFEIIuFFEIIu IIx!"LL(#$FFEIIuFFEIIuJJ MM9IIxLL(3 :"(Tr>