dE^dZddlmZddlZddlmZmZddlZddlm Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZddlZddlZddlmZddlmZmZmZmZm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+dd l,m-Z-dd l.m/Z/e rddl0m1Z1dZ2dZ3dgdZ4dZ5dZ6dhdZ7dhdZ8dhdZ9dhdZ:didZ;e djdkd$Zdod0Z?dhd1Z@dhd2ZAdpd4ZBdqd6ZCd7ZDd8ZEd9ZFedrd<ZGedsd?ZGdmdtdAZGdudEZHdFZIdvdIZJejKdwdMZLdxdQZMejNejNejOejPjQejRejSjQiZTiejNdRejOdSejRdTejUdUejVdVejNdRejWdRejXdWejYdWejZdXej[dXej\dYej]dYej^dZej_dZej`d[ejad[ejOdSejbdSejRdTejcdTejdd\ejed\ejfd]ejgd]iZhdydaZidbZjdzdfZkdS){zh Misc tools for implementing data structures Note: pandas.core.common is *not* part of the public API. ) annotationsN)abc defaultdict)partial) TYPE_CHECKINGAnyCallable Collection GeneratorHashableIterableSequencecastoverload)lib) AnyArrayLike ArrayLikeNpDtype RandomStateT)'construct_1d_object_array_from_listlike) is_array_like is_bool_dtypeis_extension_array_dtype is_integer)ABCExtensionArrayABCIndex ABCSeries)iterable_not_string)isna)Indexc#fK|D]+}t|rt|Ed{V'|V,dS)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 N/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/core/common.pyr#r#<s_" w ' ' w'' ' ' ' ' ' ' ' 'MMMM cz|dj}|ddD]"} |j|krd}#t$rd}YwxYw|S)Nr)name ValueError)objsr*objs r&consensus_name_attrr.Tse 7znot_none..s" 3 3C3?C???? 3 3r'rLargss r&not_nonerSs 4 34 3 3 33r'c4td|DS)z? Returns a boolean indicating if any argument is None. c3K|]}|duV dSrKrLrMs r&rPzany_none..&++ssd{++++++r'r<rQs r&any_nonerX! ++d+++ + ++r'c4td|DS)zA Returns a boolean indicating if all arguments are None. c3K|]}|duV dSrKrLrMs r&rPzall_none..rVr'allrQs r&all_noner^rYr'c4td|DS)zC Returns a boolean indicating if any argument is not None. c3K|]}|duV dSrKrLrMs r&rPzany_not_none..&//3s$//////r'rWrQs r& any_not_nonerb! //$/// / //r'c4td|DS)zE Returns a boolean indicating if all arguments are not None. c3K|]}|duV dSrKrLrMs r&rPzall_not_none..rar'r\rQs r& all_not_nonerfrcr'intc4td|DS)z; Returns the count of arguments that are not None. c3K|]}|duV dSrKrLrNxs r&rPz!count_not_none..s&++q}++++++r')sumrQs r&count_not_nonermrYr'.valuesArrayLike | list | tuple | zipr7NpDtype | None np.ndarraycdSrKrLrnr7s r&asarray_tuplesaferts Cr'r rcdSrKrLrss r&rtrtCr'ct|ttfs t|dst|}nt|tr|jSt|tr$|t jtfvrt|S tj 5tj dt j t j||}dddn #1swxYwYn#t$rt|cYSwxYwt!|jjt&rt j|t}|jdkrd|D}t|}|S)N __array__ignorer7c,g|]}t|SrL)tuplerjs r& z%asarray_tuplesafe..s+++q%((+++r')r4r=r}hasattrr_valuesr5r8objectrwarningscatch_warnings simplefilterVisibleDeprecationWarningr9r+ issubclassr7r?strndim)rnr7results r&rtrts ve} - -1M1Mf FH % %~&$?Ebj&-A$A$A6v>>> ?  $ & & 5 5  !(B,H I I IZe444F 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 ??? 7v>>>>> ?&,#S))2F&111 {a++F+++8@@ Ms6C4&6C( C4(C,,C4/C,0C44DDlabelsnp.ndarray | Iterablect|ttfr|g}t|ttjfs$ t|}n#t $r|g}YnwxYwt||}|S)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 rz)r4rr}r=r5r6 TypeErrorrt)rr7s r&index_labels_to_arrayrs&3,'' ftRZ0 1 1 &\\FF   XFFF vU 3 3 3F MsA A"!A"cH|t|ttfs|gS|SrK)r4r}r=r-s r&maybe_make_listr"s% z#t}==u Jr'r-Iterable[T] | TCollection[T] | Tct|tjr)t|tjst |St t |}|S)zB If obj is Iterable but not list-like, consume into list. )r4rr Sizedr=rr rs r&maybe_iterable_to_listr(sK#s|$$ZSY-G-GCyy z3  C Jr'cbt|to|jduo|jduo|jduS)z We have a null slice. Nr4slicestartstopsteprs r& is_null_slicer2sE 3  I   H   H  r'cpt|to!|jduo|jduo|j|jkS)z> We have an empty slice, e.g. no values are selected. N)r4rrrrs r&is_empty_slicer>sG 3 " IT ! " HD  " I ! r' list[bool]cd|DS)zX Find non-trivial slices in "line": return a list of booleans with same length. cXg|]'}t|tot| (SrL)r4rr)rNks r&r~z"is_true_slices..Ns2 H H HaJq% 9q)9)9%9 H H Hr'rL)r$s r&is_true_slicesrJs I H4 H H HHr'r$cjt|to|jdko|j|ko|jduS)z& We have a full length slice. rNr)r-r$s r& is_full_slicerRsD 3  IN  H   H  r'ct|drt|dSt|trt |jSt |rt|jSdS)N__name__) rgetattrr4rget_callable_namefunccallabler?rrs r&rr^sjsJ(sJ'''#w+ ***}}"Cyy!! 4r'c 6t|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_callableros2-~c,,V,,, r'c6tj|s>t|trt t|jSt |}t|tj std||tkrtd|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) inspectisclassr4rrdefault_factoryr?rrMappingr)intos r&standardize_mappingrs* ?4  dK ( ( >;(<== =Dzz dCK ( (53T33444 {IJJJ Kr'statenp.random.GeneratorcdSrKrLrs r& random_staterrvr'Gint | ArrayLike | np.random.BitGenerator | np.random.RandomState | Nonenp.random.RandomStatecdSrKrLrs r&rrs Cr'RandomState | Nonecxt|s.t|st|tjjrtj|St|tjjr|St|tjjr|S| tjStd)a 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. .. versionchanged:: 1.1.0 array-like and BitGenerator object now passed to np.random.RandomState() as seed Default None. Returns ------- np.random.RandomState or np.random.Generator. If state is None, returns np.random Nzdrandom_state must be an integer, array-like, a BitGenerator, Generator, a numpy RandomState, or None) rrr4r5random BitGeneratorrr r+rs r&rrs4 5     eRY3 4 4 "y$$U+++ E290 1 1   E29. / /  y +   r'r/Callable[..., T] | tuple[Callable[..., T], str]rct|tr*|\}}||vr|d}t||||<||i|S||g|Ri|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)r4r}r+)r-rrRrtargetmsgs r&pipers4$* f V  LLLCS// !vtT$V$$$tC)$)))&)))r'cXfd}ttjtfr|nS)zv Returns a function that will map names/labels, dependent if mapper is a dict, Series or just a function. c |vr|S|SrKrL)rkmappers r&fzget_rename_function..fs ;;!9 Hr')r4rrr)rrs` r&get_rename_functionr s>  6CK#;<< H11&Hr'"Hashable | Iterable | AnyArrayLikelist | AnyArrayLikect|ttjtt t fr|St|tjr$t|tst|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. ) r4r=r5r6rrrrr r)rns r&convert_to_list_likersa&4XyBSTUU FCL ) )*VS2I2IF|| 8Or'attrrGenerator[None, None, None]c#Kt||}t||| |Vt|||dS#t|||wxYw)zTemporarily set attribute on an object. Args: obj: Object whose attribute will be modified. attr: Attribute to modify. value: Value to temporarily set attribute to. Yields: obj with modified attribute. N)rsetattr)r-rvalue old_values r& temp_setattrr'sgT""I Cu& T9%%%%%T9%%%%s <Aindexr!Nonect|t|kr0tdt|dt|ddS)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+)datars r&require_length_matchr;sg 4yyCJJ D   E       r'rlmaxminr]r<meanprodstdvarmediancumprodcumsumrOr str | Nonec6t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetrOs r&get_cython_funcrns   S ! !!r'c8t||S)zd if we define a builtin function for this argument, return it, otherwise return the arg )_builtin_tablerrs r&is_builtin_funcrus   c3 ' ''r'namesSequence[Hashable | None]list[Hashable]c4dt|DS)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. c&g|]\}}|d|n|S)Nlevel_rL)rNir*s r&r~z&fill_missing_names..s* S S Swq$DLLQLLLd S S Sr') enumerate)rs r&fill_missing_namesr}s! T S)EBRBR S S SSr')r/rr0r1)r0r1)r0rg).)rnror7rpr0rq)rnr r7rpr0rrK)rrr7rpr0rq)r-rr0r)r0r)r$rgr0r1)rrr0r)rrr0r)rr)rrr0r)rnrr0r)rrr0r)rr!r0r)rOr r0r)rrr0r)l__doc__ __future__rbuiltins collectionsrr contextlib functoolsrrtypingrrr r r r r rrrrnumpyr5 pandas._libsrpandas._typingrrrrrpandas.core.dtypes.castrpandas.core.dtypes.commonrrrrpandas.core.dtypes.genericrrrpandas.core.dtypes.inferencerpandas.core.dtypes.missingr pandasr!r#r.rCrHrSrXr^rbrfrmrtrrrrrrrrrrrrrrcontextmanagerrrrlrmaximumreducerminimumrr]r<nansumrnanmeanrnanprodrnanstdrnanvarr nanmediannanmaxnanminr nancumprodr nancumsumrrrrrLr'r&rsz #"""""                        LKKKKK  =<<<<<++++++04444n,444,,,,,,,,00000000,,,, DG   B<@:         IIII    ""@     5 5 5 5 5 p"*"*"*"*J I I I     &&&&&     " L"& L"*# L"*#  L% L% L%FE  FE  FE IuGVJGVJFEIuFEIu Ix!"L(#$FEIuFEIuJ M9IxL(3 :""""(((TTTTTTr'