o Ti3C@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"R/var/www/html/evchargy.com/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)Zobjsr%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)ZskipnaFr) isinstancernpndarrayrrdtypeobject_asarrayrZ is_bool_arrayr&rlistlentypeZ is_bool_list)r)Z key_arrayZna_msgr"r"r#is_bool_indexer`s$       r5cCst|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  r8cGdd|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_nonerAcGtdd|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_nonerHcGrC)zA Returns a boolean indicating if all arguments are None. csrDr:r"r;r"r"r#r>rEzall_none..allr?r"r"r#all_nonerIrLcGrC)zC Returns a boolean indicating if any argument is not None. cs|]}|duVqdSr:r"r;r"r"r#r>rEzany_not_none..rFr?r"r"r# any_not_nonerIrNcGrC)zE Returns a boolean indicating if all arguments are not None. csrMr:r"r;r"r"r#r>rEzall_not_none..rJr?r"r"r# all_not_nonerIrOintcGrC)z; Returns the count of arguments that are not None. csrMr:r"r<xr"r"r#r>rEz!count_not_none..)sumr?r"r"r#count_not_nonerIrT.valuesArrayLike | list | tuple | zipr/NpDtype | None np.ndarraycCdSr:r"rUr/r"r"r#asarray_tuplesafesr[rrcCrYr:r"rZr"r"r#r[cCst|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)NZ __array__ignorer/cSsg|]}t|qSr")tuplerQr"r"r# sz%asarray_tuplesafe..)r,r2r`hasattrrZ_valuesr-r0objectrwarningscatch_warningsr simplefilterZVisibleDeprecationWarningr1r& issubclassr/r4strndim)rUr/resultr"r"r#r[s,      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,rhr`r2r-r. TypeErrorr[)rkr/r"r"r#index_labels_to_arrays    rncCs |durt|ttfs|gS|Sr:)r,r`r2r'r"r"r#maybe_make_list"srpr'Iterable[T] | TCollection[T] | TcCs.t|tjrt|tjst|Stt|}|S)zB If obj is Iterable but not list-like, consume into list. )r,rrSizedr2rrror"r"r#maybe_iterable_to_list(s rtcCs(t|to|jduo|jduo|jduS)z We have a null slice. Nr,slicestartstopstepror"r"r# is_null_slice2 rzcCs*t|to|jduo|jduo|j|jkS)z> We have an empty slice, e.g. no values are selected. N)r,rvrwrxror"r"r#is_empty_slice>s  r| list[bool]cCr9)zX Find non-trivial slices in "line": return a list of booleans with same length. cSs g|] }t|to t| qSr")r,rvrz)r<kr"r"r#raNs z"is_true_slices..r")r r"r"r#is_true_slicesJrBrr cCs(t|to|jdko|j|ko|jduS)z& We have a full length slice. rNru)r'r r"r"r# is_full_sliceRr{rcCs>t|dr t|dSt|trt|jSt|rt|jSdS)N__name__) rbgetattrr,r get_callable_namefunccallabler4rror"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)Zmaybe_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_factoryr4rgrMappingrm)Zintor"r"r#standardize_mappings    rstatenp.random.GeneratorcCrYr:r"rr"r"r# random_stater\rHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Nonenp.random.RandomStatecCrYr: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,r`r&)r'rr@rtargetmsgr"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")rRmapperr"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,r2r-r.rrrrrrh)rUr"r"r#convert_to_list_likes rTattrrh 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)r3r&)datarr"r"r#require_length_match,srznp.sumznp.maximum.reduceznp.minimum.reducerSmaxminrKrGmeanprodstdvarmediancumprodcumsumr=r str | NonecCs t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetr=r"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)NZlevel_r")r<ir%r"r"r#ras&z&fill_missing_names..) enumerate)rr"r"r#fill_missing_namesusr)r)r r*r+)r*r+)r*rP).)rUrVr/rWr*rX)rUrr/rWr*rr:)rkrlr/rWr*rX)r'rqr*rr)r*r})r rPr*r+)rrr*r)rrr*r)rr)rrr*r)rUrr*r)T)rrhrr+r*r)rrr*r)r=r r*r)rrr*r)l__doc__ __future__rbuiltins collectionsrrcollections.abcrrrrr contextlib functoolsr rtypingr r r rrrdnumpyr-Z pandas._libsrZpandas.compat.numpyrZpandas.core.dtypes.castrZpandas.core.dtypes.commonrrZpandas.core.dtypes.genericrrrZpandas.core.dtypes.inferencerZpandas._typingrrrrrZpandasrrr(r5r8rArHrLrNrOrTr[rnrprtrzr|rrrrrrrrrcontextmanagerrrrSrmaximumreducerminimumrZ_builtin_table_aliasrKrGZnansumrZnanmeanrZnanprodrZnanstdrZnanvarrZ nanmedianZnanmaxZnanminrZ nancumprodrZ nancumsumrrrrr"r"r"r#s        5       "        #%