o iM@srdZddlmZddlZddlmZmZddlZddlm Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZddlZddlZddlmZddlmZmZmZmZm Z dd l!m"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/ddl0m1Z1e rddl2m3Z3ddZ4ddZ5dddZ6ddddZ7ddZ8dd d!Z9dd"d#Z:dd$d%Z;dd&d'Zeddd5d2Z>ddd6d2Z> ddd9d:Z?d;d<Z@dd@dAZAddBdCZBddEdFZCddHdIZDdJdKZEdLdMZFdNdOZGeddRdSZHeddVdSZHdddXdSZHdd\d]ZId^d_ZJddbdcZKejLddgdhZMddldmZNejOejOejPejQjRejSejTjRiZUiejOdnejPdoejSdpejVdqejWdrejOdnejXdnejYdsejZdsej[dtej\dtej]duej^duej_dvej`dvejadwejbdwejPdoejcdoejSdpejddpejedxejfdxejgdyejhdyiZidd}d~ZjddZkdddZldddZm ddddZndS)zh Misc tools for implementing data structures Note: pandas.core.common is *not* part of the public API. ) annotationsN)abc defaultdict)partial) TYPE_CHECKINGAnyCallable CollectionHashableIterableIteratorSequencecastoverload)lib) AnyArrayLike ArrayLikeNpDtype RandomStateT)find_stack_level)'construct_1d_object_array_from_listlike) is_array_like is_bool_dtypeis_extension_array_dtype is_integer)ABCExtensionArrayABCIndex ABCSeries)iterable_not_string)isna)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%M/var/www/edux/Edux_v2/venv/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)objsr(objr%r%r&consensus_name_attrUs   r,keyrreturnboolcCst|ttjtfst|rBt|jrB|jtjkr9t |}t |s7d}t |dkr5t |r5t|dSdSt|jr@dSdSt|tr\t|dkr\t|turWt|}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 valuesbooleanFTr) isinstancernpndarrayrrrdtypeobject_asarrayr is_bool_array infer_dtyper anyr)rlistlentype is_bool_list)r- key_arrayna_msgr%r%r&is_bool_indexer`s,       r@F warn_floatcCs4t|r|r|rtjdttdt|S|S)a To avoid numpy DeprecationWarnings, cast float to integer where valid. Parameters ---------- val : scalar warn_float : bool, default False If True, issue deprecation warning for a float indexer. Returns ------- outval : scalar zIndexing with a float is deprecated, and will raise an IndexError in pandas 2.0. You can manually convert to an integer key instead. stacklevel)ris_floatrwarningswarn FutureWarningrint)valrAr%r%r&cast_scalar_indexersrJcGdd|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_nonerScGtdd|DS)z? Returns a boolean indicating if any argument is None. cs|]}|duVqdSrLr%rMr%r%r&rPzany_none..r9rQr%r%r&any_nonerYcGrU)zA Returns a boolean indicating if all arguments are None. csrVrLr%rMr%r%r&rPrWzall_none..allrQr%r%r&all_nonerZr]cGrU)zC Returns a boolean indicating if any argument is not None. cs|]}|duVqdSrLr%rMr%r%r&rPrWzany_not_none..rXrQr%r%r& any_not_nonerZr_cGrU)zE Returns a boolean indicating if all arguments are not None. csr^rLr%rMr%r%r&rPrWzall_not_none..r[rQr%r%r& all_not_nonerZr`rHcGrU)z; Returns the count of arguments that are not None. csr^rLr%rNxr%r%r&rPrWz!count_not_none..)sumrQr%r%r&count_not_nonerZrd.valuesArrayLike | list | tuple | zipr4NpDtype | None np.ndarraycCdSrLr%rer4r%r%r&asarray_tuplesafesrkr rcCrirLr%rjr%r%r&rkcCst|ttfst|dst|}nt|tr|jSt|tr)|tjtfvr)t |Sz$t t dtj tj||d}Wdn1sGwYWn tyZt |YSwt|jjtritj|td}|jdkrydd|D}t |}|S)N __array__ignorer4cSsg|]}t|qSr%)tuplerar%r%r& sz%asarray_tuplesafe..)r1r:rqhasattrr_valuesr2r5objectrrEcatch_warnings simplefilterVisibleDeprecationWarningr6r) issubclassr4r<strndim)rer4resultr%r%r&rks*      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 ro)r1rzrqr:r2r3 TypeErrorrk)r}r4r%r%r&index_labels_to_array s    rcCs |durt|ttfs|gS|SrL)r1rqr:r+r%r%r&maybe_make_list)srr+Iterable[T] | TCollection[T] | TcCs.t|tjrt|tjst|Stt|}|S)zB If obj is Iterable but not list-like, consume into list. )r1rr Sizedr:rr rr%r%r&maybe_iterable_to_list/s rcCs(t|to|jduo|jduo|jduS)z We have a null slice. Nr1slicestartstopsteprr%r%r& is_null_slice9 r list[bool]cCrK)zX Find non-trivial slices in "line": return a list of booleans with same length. cSs g|] }t|to t| qSr%)r1rr)rNkr%r%r&rrIs z"is_true_slices..r%)r#r%r%r&is_true_slicesErTrr#cCs(t|to|jdko|j|ko|jduS)z& We have a full length slice. rNr)r+r#r%r%r& is_full_sliceMrrcCs>t|dr t|dSt|trt|jSt|rt|jSdS)N__name__) rsgetattrr1rget_callable_namefunccallabler<rrr%r%r&rYs     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_callablejs 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) inspectisclassr1rrdefault_factoryr<ryrMappingr)intor%r%r&standardize_mapping{s    rstatenp.random.GeneratorcCrirLr%rr%r%r& random_staterlrGint | ArrayLike | np.random.BitGenerator | np.random.RandomState | Nonenp.random.RandomStatecCrirLr%rr%r%r&rsRandomState | NonecCsdt|st|st|tjjrtj|St|tjjr|St|tjjr'|S|dur.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) rrr1r2random BitGeneratorr Generatorr)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)r1rqr))r+rrRrtargetmsgr%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|SrLr%)rbmapperr%r&f szget_rename_function..f)r1rrr)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. ) r1r:r2r3rrrrr rz)rer%r%r&convert_to_list_likes rattrrzIterator[None]c cs@t||}t|||z |VWt|||dSt|||w)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_valuer%r%r& temp_setattr"s  rindexr!NonecCs2t|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)r;r))datarr%r%r&require_length_match6srrcmaxminr\r9meanprodstdvarmediancumprodcumsumrOr str | NonecCs t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetrOr%r%r&get_cython_funcis 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_funcps 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%)rNir(r%r%r&rrs&z&fill_missing_names..) enumerate)rr%r%r&fill_missing_namesxsr numeric_onlybool | None | lib.NoDefaultcCs*|tjur d}|S|durd}|S|}|S)aDetermine the Boolean value of numeric_only. See GH#46560 for details on the deprecation. Parameters ---------- numeric_only : bool, None, or lib.no_default Value passed to the method. Returns ------- Resolved value of numeric_only. TNF)r no_default)rr|r%r%r&resolve_numeric_onlys rclsr<r(deprecate_nonecCsf|dvrd}nd}d|d|jd|d}|r |d|d 7}|d |d 7}tj|ttd d S)aEmit FutureWarning message for deprecation of numeric_only. See GH#46560 for details on the deprecation. Parameters ---------- cls : type pandas type that is generating the warning. name : str Name of the method that is generating the warning. deprecate_none : bool, default False Whether to also warn about the deprecation of specifying ``numeric_only=None``. )r\r9 bool_onlyrzThe default value of z in .z? is deprecated. In a future version, it will default to False. zIn addition, specifying 'z=None' is deprecated. z2Select only valid columns or specify the value of z to silence this warning.rBN)rrErFrGr)rr(rarg_namerr%r%r&deprecate_numeric_only_defaults r)r-rr.r/)F)rAr/)r.r/)r.rH).)rerfr4rgr.rh)rer r4rgr.rrL)r}r~r4rgr.rh)r+rr.r)r.r)r#rHr.r/)rrr.r)rrr.r)rr)rrr.r)rerr.r)rrzr.r)rr!r.r)rOrr.r)rrr.r)rrr.r/)rr<r(rzrr/r.r)o__doc__ __future__rbuiltins collectionsrr contextlib functoolsrrtypingrrrr r r r r rrrEnumpyr2 pandas._libsrpandas._typingrrrrrpandas.util._exceptionsrpandas.core.dtypes.castrpandas.core.dtypes.commonrrrrpandas.core.dtypes.genericrrrpandas.core.dtypes.inferencerpandas.core.dtypes.missingr pandasr!r"r,r@rJrSrYr]r_r`rdrkrrrrrrrrrrrrrcontextmanagerrrrcrmaximumreducerminimumrr\r9nansumrnanmeanrnanprodrnanstdrnanvarr nanmediannanmaxnanminr nancumprodr nancumsumrrrrrrr%r%r%r&s  0        7       "        8%