o ҷh3C@sndZddlmZddlZddlmZmZddlmZm 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&dd l'm(Z(erddl)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddZ1ddZ2dddZ3ddZ4ddZ5dddZ6dd d!Z7dd"d#Z8dd$d%Z9dd'd(Z:e )ddd/d0Z;eddd3d0Z;ddd4d0Z; ddd7d8Zd?Z>dd@dAZ?ddBdCZ@ddEdFZAddHdIZBdJdKZCdLdMZDdNdOZEeddRdSZFeddVdSZFdddXdSZFdd\d]ZGd^d_ZHddbdcZIe jJ ddddidjZKddndoZLejMejMejNejOjPejQejRjPiZSejMdpejNdqejQdriZTiejMdsejNdtejQduejUdvejVdwejMdsejWdsejXdxejYdxejZdyej[dyej\dzej]dzej^d{ej_d{ej`d|ejad|ejNdtejbdtejQduejcduejdd}ejed}ejfd~ejgd~iZhdddZiddZjdddZkdS)zh 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 ABCSeries)iterable_not_string) AnyArrayLike ArrayLikeNpDtype RandomStateT)Indexccs.|D]}t|rt|EdHq|VqdS)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)lineelementr"E/var/www/html/venv/lib/python3.10/site-packages/pandas/core/common.pyr=s rc CsH|dj}|ddD]}z |j|krd}Wq ty!d}Yq w|S)Nr)name ValueError)objsr%objr"r"r#consensus_name_attrUs   r)keyr returnboolcCst|ttjttfr4|jtjkr+t|}t |s)d}t j |ddr't |dSdSt |jr2dSdSt|t rNt|dkrNt|t urIt |}t |SdS)a Check whether `key` is a valid boolean indexer. Parameters ---------- key : Any Only list-likes may be considered boolean indexers. All other types are not considered a boolean indexer. For array-like input, boolean ndarrays or ExtensionArrays with ``_is_boolean`` set are considered boolean indexers. Returns ------- bool Whether `key` is a valid boolean indexer. Raises ------ ValueError When the array is an object-dtype ndarray or ExtensionArray and contains missing values. See Also -------- check_array_indexer : Check that `key` is a valid array to index, and convert to an ndarray. z=Cannot mask with non-boolean array containing NA / NaN valuesT)skipnaFr) isinstancernpndarrayrrdtypeobject_asarrayr is_bool_arrayr&rlistlentype is_bool_list)r* key_arrayna_msgr"r"r#is_bool_indexer`s$       r;cCst|r |r td|S)z Disallow indexing with a float key, even if that key is a round number. Parameters ---------- val : scalar Returns ------- outval : scalar zYIndexing with a float is no longer supported. Manually convert to an integer key instead.)ris_floatr IndexError)valr"r"r#cast_scalar_indexers  r?cGdd|DS)zL Returns a generator consisting of the arguments that are not None. css|] }|dur|VqdSNr".0argr"r"r# sznot_none..r"argsr"r"r#not_nonerHcGtdd|DS)z? Returns a boolean indicating if any argument is None. cs|]}|duVqdSrAr"rBr"r"r#rEzany_none..anyrFr"r"r#any_nonerOcGrJ)zA Returns a boolean indicating if all arguments are None. csrKrAr"rBr"r"r#rErLzall_none..allrFr"r"r#all_nonerPrScGrJ)zC Returns a boolean indicating if any argument is not None. cs|]}|duVqdSrAr"rBr"r"r#rErLzany_not_none..rMrFr"r"r# any_not_nonerPrUcGrJ)zE Returns a boolean indicating if all arguments are not None. csrTrAr"rBr"r"r#rErLzall_not_none..rQrFr"r"r# all_not_nonerPrVintcGrJ)z; Returns the count of arguments that are not None. csrTrAr"rCxr"r"r#rErLz!count_not_none..)sumrFr"r"r#count_not_nonerPr[.valuesArrayLike | list | tuple | zipr1NpDtype | None np.ndarraycCdSrAr"r\r1r"r"r#asarray_tuplesafesrbrrcCr`rAr"rar"r"r#rbcCst|ttfst|dst|}nt|tr|jSt|tr)|tjtfvr)t |Sz&t t s8t dtjtj||d}Wdn1sIwYWn ty\t |YSwt|jjtrktj|td}|jdkr{dd|D}t |}|S)N __array__ignorer1cSsg|]}t|qSr")tuplerXr"r"r# sz%asarray_tuplesafe..)r.r5rhhasattrr_valuesr/r2objectrwarningscatch_warningsr simplefilterVisibleDeprecationWarningr3r& issubclassr1r7strndim)r\r1resultr"r"r#rbs,      labelsnp.ndarray | IterablecCsZt|ttfr |g}t|ttjfs%zt|}Wn ty$|g}Ynwt||d}|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 rf)r.rrrhr5r/r0 TypeErrorrb)rur1r"r"r#index_labels_to_arrays    rxcCs |durt|ttfs|gS|SrA)r.rhr5r(r"r"r#maybe_make_list"srzr(Iterable[T] | TCollection[T] | TcCs.t|tjrt|tjst|Stt|}|S)zB If obj is Iterable but not list-like, consume into list. )r.rrSizedr5rrryr"r"r#maybe_iterable_to_list(s r~cCs(t|to|jduo|jduo|jduS)z We have a null slice. Nr.slicestartstopstepryr"r"r# is_null_slice2 rcCs*t|to|jduo|jduo|j|jkS)z> We have an empty slice, e.g. no values are selected. N)r.rrrryr"r"r#is_empty_slice>s  r list[bool]cCr@)zX Find non-trivial slices in "line": return a list of booleans with same length. cSs g|] }t|to t| qSr")r.rr)rCkr"r"r#riNs z"is_true_slices..r")r r"r"r#is_true_slicesJrIrr cCs(t|to|jdko|j|ko|jduS)z& We have a full length slice. rNr)r(r r"r"r# is_full_sliceRrrcCs>t|dr t|dSt|trt|jSt|rt|jSdS)N__name__) rjgetattrr.r get_callable_namefunccallabler7rryr"r"r#r^s     rcKst|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(kwargsr"r"r#apply_if_callableos rcCsVt|st|trtt|jSt|}t|tj s!t d||tkr)t d|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_factoryr7rqrMappingrw)intor"r"r#standardize_mappings    rstatenp.random.GeneratorcCr`rAr"rr"r"r# random_statercrHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Nonenp.random.RandomStatecCr`rAr"rr"r"r#rsRandomState | NonecCsbt|st|tjtjjfrtj|St|tjjr|St|tjjr&|S|dur-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 Nzdrandom_state must be an integer, array-like, a BitGenerator, Generator, a numpy RandomState, or None) rr.r/r0random BitGeneratorrrr&rr"r"r#rs r/Callable[..., T] | tuple[Callable[..., T], str]rcOsXt|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)r.rhr&)r(rrGrtargetmsgr"r"r#pipes  rcs$fdd}ttjtfr|SS)zv Returns a function that will map names/labels, dependent if mapper is a dict, Series or just a function. cs|vr|S|SrAr")rYmapperr"r#fszget_rename_function..f)r.rrr)rrr"rr#get_rename_functions r"Hashable | Iterable | AnyArrayLikelist | AnyArrayLikecCs>t|ttjtttfr |St|tjrt|t st|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.r5r/r0rrrrrrr)r\r"r"r#convert_to_list_likes rTattrrr conditionGenerator[None, None, None]c csR|rt||}t|||z|VW|rt|||dSdS|r(t|||ww)a~Temporarily set attribute on an object. Args: obj: Object whose attribute will be modified. attr: Attribute to modify. value: Value to temporarily set attribute to. condition: Whether to set the attribute. Provided in order to not have to conditionally use this context manager. Yields: obj with modified attribute. N)rsetattr)r(rvaluer old_valuer"r"r# temp_setattrs  rindexrNonecCs2t|t|krtdt|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)r6r&)datarr"r"r#require_length_match,srznp.sumznp.maximum.reduceznp.minimum.reducerZmaxminrRrNmeanprodstdvarmediancumprodcumsumrDr str | NonecCs t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetrDr"r"r#get_cython_funcfs rcCs t||S)zd if we define a builtin function for this argument, return it, otherwise return the arg )_builtin_tablerrr"r"r#is_builtin_funcms rnamesSequence[Hashable | None]list[Hashable]cCsddt|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. cSs&g|]\}}|durd|n|qS)Nlevel_r")rCir%r"r"r#ris&z&fill_missing_names..) enumerate)rr"r"r#fill_missing_namesusr)r*r r+r,)r+r,)r+rW).)r\r]r1r^r+r_)r\rr1r^r+rrA)rurvr1r^r+r_)r(r{r+r|)r+r)r rWr+r,)rrr+r)rrr+r)rr)rrr+r)r\rr+r)T)rrrrr,r+r)rrr+r)rDr r+r)rrr+r)l__doc__ __future__rbuiltins collectionsrrcollections.abcrrrrr contextlib functoolsr rtypingr r r rrrmnumpyr/ pandas._libsrpandas.compat.numpyrpandas.core.dtypes.castrpandas.core.dtypes.commonrrpandas.core.dtypes.genericrrrpandas.core.dtypes.inferencerpandas._typingrrrrrpandasrrr)r;r?rHrOrSrUrVr[rbrxrzr~rrrrrrrrrrrcontextmanagerrrrZrmaximumreducerminimumr_builtin_table_aliasrRrNnansumrnanmeanrnanprodrnanstdrnanvarr nanmediannanmaxnanminr nancumprodr nancumsumrrrrr"r"r"r#s        5       "        #%