a  d3C2@sdZddlmZddlZddlmZmZddlmZm 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/m0Z0ddZ1ddZ2dddddZ3ddZ4ddZ5ddddZ6ddd d!Z7ddd"d#Z8ddd$d%Z9d&dd'd(Z:edd*d+d,d-d.d/Z;edd0d+d1d-d2d/Z;dd0d+d1d-d3d/Z;dd4d+d,d5d6d7ZZ>ddd?d@Z?dddAdBZ@dCddDdEZAd&ddFdGdHZBdIdJZCdKdLZDdMdNZEedOdOdPdQdRZFedSdTdPdUdRZFddVdWdXdRZFdYdZd[d\d]ZGd^d_ZHd`dadbdcddZIe jJddfddgdhdidjZKdkdldmdndoZLejMejMejNejOjPejQejRjPiZSejMdpejNdqejQdriZTejMdsejNdtejQduejUdvejVdwejMdsejWdsejXdxejYdxejZdyej[dyej\dzej]dzej^d{ej_d{ej`d|ejad|ejNdtejbdtejQduejcduejdd}ejed}ejfd~ejgd~iZhdddddZiddZjdddddZkdS)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|r t|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"[/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pandas/core/common.pyr=src CsJ|dj}|ddD].}z|j|kr*d}WqtyBd}Yq0q|S)Nr)name ValueError)objsr%objr"r"r#consensus_name_attrUs    r)r bool)keyreturncCst|ttjttfrf|jtjkrVt|}t |sRd}t j |ddrNt |dSdSt |jrdSn4t|t rt|dkrt|t urt |}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"        r6cCst|r|rtd|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 r9cGsdd|DS)zL Returns a generator consisting of the arguments that are not None. css|]}|dur|VqdSNr".0argr"r"r# znot_none..r"argsr"r"r#not_nonesrB)r,cGstdd|DS)z? Returns a boolean indicating if any argument is None. css|]}|duVqdSr:r"r;r"r"r#r>r?zany_none..anyr@r"r"r#any_nonesrEcGstdd|DS)zA Returns a boolean indicating if all arguments are None. css|]}|duVqdSr:r"r;r"r"r#r>r?zall_none..allr@r"r"r#all_nonesrHcGstdd|DS)zC Returns a boolean indicating if any argument is not None. css|]}|duVqdSr:r"r;r"r"r#r>r?zany_not_none..rCr@r"r"r# any_not_nonesrIcGstdd|DS)zE Returns a boolean indicating if all arguments are not None. css|]}|duVqdSr:r"r;r"r"r#r>r?zall_not_none..rFr@r"r"r# all_not_nonesrJintcGstdd|DS)z; Returns the count of arguments that are not None. css|]}|duVqdSr:r"r<xr"r"r#r>r?z!count_not_none..)sumr@r"r"r#count_not_nonesrO.zArrayLike | list | tuple | zipzNpDtype | Nonez np.ndarray)valuesr0r,cCsdSr:r"rPr0r"r"r#asarray_tuplesafesrRrrcCsdSr:r"rQr"r"r#rRscCst|ttfs"t|ds"t|}nt|tr2|jSt|trR|tjtfvrRt |SzLt 0t spt dtjtj||d}Wdn1s0YWntyt |YS0t|jjtrtj|td}|jdkrdd|D}t |}|S)NZ __array__ignorer0cSsg|] }t|qSr")tuplerLr"r"r# r?z%asarray_tuplesafe..)r-r3rVhasattrrZ_valuesr.r1objectrwarningscatch_warningsr simplefilterZVisibleDeprecationWarningr2r& issubclassr0r5strndim)rPr0resultr"r"r#rRs&   0  znp.ndarray | Iterable)labelsr0r,cCsZt|ttfr|g}t|ttjfsJz t|}WntyH|g}Yn0t||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 rT)r-r^rVr3r.r/ TypeErrorrR)rar0r"r"r#index_labels_to_arrays    rccCs |durt|ttfs|gS|Sr:)r-rVr3r(r"r"r#maybe_make_list"srezIterable[T] | TzCollection[T] | T)r(r,cCs.t|tjr t|tjs t|Stt|}|S)zB If obj is Iterable but not list-like, consume into list. )r-rrSizedr3rrrdr"r"r#maybe_iterable_to_list(s rgcCs(t|to&|jduo&|jduo&|jduS)z We have a null slice. Nr-slicestartstopsteprdr"r"r# is_null_slice2s rmcCs*t|to(|jduo(|jduo(|j|jkS)z> We have an empty slice, e.g. no values are selected. N)r-rirjrkrdr"r"r#is_empty_slice>s  rnz list[bool]cCsdd|DS)zX Find non-trivial slices in "line": return a list of booleans with same length. cSs g|]}t|tot| qSr")r-rirm)r<kr"r"r#rWNr?z"is_true_slices..r")r r"r"r#is_true_slicesJsrp)r r,cCs(t|to&|jdko&|j|ko&|jduS)z& We have a full length slice. rNrh)r(r r"r"r# is_full_sliceRs rqcCs>t|drt|dSt|tr(t|jSt|r:t|jSdS)N__name__) rXgetattrr-r get_callable_namefunccallabler5rrrdr"r"r#rt^s     rtcKst|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 )rv)Zmaybe_callabler(kwargsr"r"r#apply_if_callableos rxcCsVt|s(t|tr tt|jSt|}t|tj sBt d||tkrRt 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_factoryr5r]rMappingrb)Zintor"r"r#standardize_mappings    r}znp.random.Generator)stater,cCsdSr:r"r~r"r"r# random_statesrzHint | np.ndarray | np.random.BitGenerator | np.random.RandomState | Noneznp.random.RandomStatecCsdSr:r"rr"r"r#rszRandomState | NonercCsft|st|tjtjjfr(tj|St|tjjr:|St|tjjrL|S|durZtjStddS)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 z/Callable[..., T] | tuple[Callable[..., T], str]r)rur,cOs\t|trB|\}}||vr,|d}t||||<||i|S||g|Ri|SdS)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 argumentN)r-rVr&)r(rurArwtargetmsgr"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|SdSr:r")rMmapperr"r#fszget_rename_function..f)r-rr|r)rrr"rr#get_rename_functions rz"Hashable | Iterable | AnyArrayLikezlist | AnyArrayLike)rPr,cCs>t|ttjtttfr|St|tjr8t|t s8t|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/rrrrrr^)rPr"r"r#convert_to_list_likes rTr^zGenerator[None, None, None])attr conditionr,c csL|rt||}t|||z|VW|rHt|||n|rFt|||0dS)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)rssetattr)r(rvaluer old_valuer"r"r# temp_setattrs  rrNone)indexr,cCs2t|t|kr.tdt|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_match,srznp.sumznp.maximum.reduceznp.minimum.reducerNmaxminrGrDmeanprodstdvarmediancumprodcumsumr z str | None)r=r,cCs t|S)zH if we define an internal function for this argument, return it ) _cython_tablegetr=r"r"r#get_cython_funcfsrcCs t||S)zd if we define a builtin function for this argument, return it, otherwise return the arg )_builtin_tablerrr"r"r#is_builtin_funcmsrzSequence[Hashable | None]zlist[Hashable])namesr,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#rWr?z&fill_missing_names..) enumerate)rr"r"r#fill_missing_namesusr).).)N)N)N)T)l__doc__ __future__rbuiltins collectionsrrcollections.abcrrrrr contextlib functoolsr rytypingr r r rrrZnumpyr.Z pandas._libsrZpandas.compat.numpyrZpandas.core.dtypes.castrZpandas.core.dtypes.commonrrZpandas.core.dtypes.genericrrrZpandas.core.dtypes.inferencerZpandas._typingrrrrrZpandasrrr)r6r9rBrErHrIrJrOrRrcrergrmrnrprqrtrxr}rrrrcontextmanagerrrrNrmaximumreducerZminimumrZ_builtin_table_aliasrGrDZnansumrZnanmeanrZnanprodrZnanstdrZnanvarrZ nanmedianZnanmaxZnanminrZ nancumprodrZ nancumsumrrrrr"r"r"r#s        5 "     #%