o թZh)D@srdZddlmZddlZddlmZmZddlmZm 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/ddl0m1Z1ddZ2ddZ3dddZ4ddZ5ddZ6dddZ7dd d!Z8dd"d#Z9dd$d%Z:dd'd(Z;e )ddd/d0Zdd>d?Z?dd@dAZ@ddBdCZAddEdFZBddHdIZCdJdKZDdLdMZEdNdOZFeddRdSZGeddVdSZGdddXdSZGdd\d]ZHd^d_ZIddbdcZJe jK ddddidjZLddndoZMejNejNejOejPjQejRejSjQiZTejNdpejOdqejRdriZUiejNdsejOdtejRduejVdvejWdwejNdsejXdsejYdxejZdxej[dyej\dyej]dzej^dzej_d{ej`d{ejad|ejbd|ejOdtejcdtejRduejdduejed}ejfd}ejgd~ejhd~iZidddZjddZkdddZldS)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 ABCMultiIndex 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#I/var/www/html/lang_env/lib/python3.10/site-packages/pandas/core/common.pyr >s r c CsH|dj}|ddD]}z |j|krd}Wq ty!d}Yq w|S)Nr)name ValueError)Zobjsr&objr#r#r$consensus_name_attrVs   r)keyr returnboolcCst|ttjttfr9t|ts9|jtjkr0t |}t |s.d}t j |ddr,t |dSdSt |jr7dSdSt|trSt|dkrSt|turNt|}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)ZskipnaFr) isinstancernpndarrayrrrdtypeobject_asarrayrZ is_bool_arrayr'rlistlentypeZ is_bool_list)r*Z key_arrayZna_msgr#r#r$is_bool_indexeras,       r6cCst|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.)rZis_floatr IndexError)valr#r#r$cast_scalar_indexers  r9cGdd|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_nonerBcGtdd|DS)z? Returns a boolean indicating if any argument is None. cs|]}|duVqdSr;r#r<r#r#r$r?zany_none..anyr@r#r#r$any_nonerIcGrD)zA Returns a boolean indicating if all arguments are None. csrEr;r#r<r#r#r$r?rFzall_none..allr@r#r#r$all_nonerJrMcGrD)zC Returns a boolean indicating if any argument is not None. cs|]}|duVqdSr;r#r<r#r#r$r?rFzany_not_none..rGr@r#r#r$ any_not_nonerJrOcGrD)zE Returns a boolean indicating if all arguments are not None. csrNr;r#r<r#r#r$r?rFzall_not_none..rKr@r#r#r$ all_not_nonerJrPintcGrD)z; Returns the count of arguments that are not None. csrNr;r#r=xr#r#r$r?rFz!count_not_none..)sumr@r#r#r$count_not_nonerJrU.valuesArrayLike | list | tuple | zipr0NpDtype | None np.ndarraycCdSr;r#rVr0r#r#r$asarray_tuplesafesr\rrcCrZr;r#r[r#r#r$r\cCs t|ttfst|dst|}nt|tr|jSt|tr!|jSt|tr1|tjt fvr1t |Sz&t t s@t dtjtj||d}Wdn1sQwYWn tydt |YSwt|jjtrstj|t d}|jdkrdd|D}t |}|S)NZ __array__ignorer0cSsg|]}t|qSr#)tuplerRr#r#r$ sz%asarray_tuplesafe..)r-r3rahasattrrZ_valuesrr.r1objectrwarningscatch_warningsr simplefilterZVisibleDeprecationWarningr2r' issubclassr0r5strndim)rVr0resultr#r#r$r\s0       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 r_)r-rirar3r.r/ TypeErrorr\)rlr0r#r#r$index_labels_to_array s    rocCs |durt|ttfs|gS|Sr;)r-rar3r(r#r#r$maybe_make_list'srqr(Iterable[T] | TCollection[T] | TcCs.t|tjrt|tjst|Stt|}|S)zB If obj is Iterable but not list-like, consume into list. )r-rrSizedr3rrrpr#r#r$maybe_iterable_to_list-s rucCs(t|to|jduo|jduo|jduS)z We have a null slice. Nr-slicestartstopsteprpr#r#r$ is_null_slice7 r{cCs*t|to|jduo|jduo|j|jkS)z> We have an empty slice, e.g. no values are selected. N)r-rwrxryrpr#r#r$is_empty_sliceCs  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-rwr{)r=kr#r#r$rbSs z"is_true_slices..r#)r!r#r#r$is_true_slicesOrCrr!cCs(t|to|jdko|j|ko|jduS)z& We have a full length slice. rNrv)r(r!r#r#r$ is_full_sliceWr|rcCs>t|dr t|dSt|trt|jSt|rt|jSdS)N__name__) rcgetattrr-r get_callable_namefunccallabler5rrpr#r#r$rcs     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)Zmaybe_callabler(kwargsr#r#r$apply_if_callablets 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_factoryr5rhrMappingrn)Zintor#r#r$standardize_mappings    rstatenp.random.GeneratorcCrZr;r#rr#r#r$ random_stater]rHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Nonenp.random.RandomStatecCrZr;r#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.r/randomZ 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-rar')r(rrArtargetmsgr#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|Sr;r#)rSmapperr#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-r3r.r/rrrrrri)rVr#r#r$convert_to_list_likes rTattrri conditionGenerator[None, None, None]c csR|rt||}t|||z|VW|rt|||dSdS|r(t|||ww)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(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)r4r')datarr#r#r$require_length_match8srznp.sumznp.maximum.reduceznp.minimum.reducerTmaxminrLrHmeanprodstdvarmediancumprodcumsumr>r str | NonecCs t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetr>r#r#r$get_cython_funcrs 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_funcys 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)NZlevel_r#)r=ir&r#r#r$rbs&z&fill_missing_names..) enumerate)rr#r#r$fill_missing_namessr)r*r r+r,)r+r,)r+rQ).)rVrWr0rXr+rY)rVrr0rXr+rr;)rlrmr0rXr+rY)r(rrr+rs)r+r~)r!rQr+r,)rrr+r)rrr+r)rr)rrr+r)rVrr+r)T)rrirr,r+r)rrr+r)r>r r+r)rrr+r)m__doc__ __future__rbuiltins collectionsrrcollections.abcrrrrr contextlib functoolsr rtypingr r r rrrenumpyr.Z pandas._libsrZpandas.compat.numpyrZpandas.core.dtypes.castrZpandas.core.dtypes.commonrrZpandas.core.dtypes.genericrrrrZpandas.core.dtypes.inferencerZpandas._typingrrrrrZpandasrr r)r6r9rBrIrMrOrPrUr\rorqrur{r}rrrrrrrrrcontextmanagerrrrTrmaximumreducerminimumrZ_builtin_table_aliasrLrHZnansumrZnanmeanrZnanprodrZnanstdrZnanvarrZ nanmedianZnanmaxZnanminrZ nancumprodrZ nancumsumrrrrr#r#r#r$s        7       $        #%