9i8dZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z m Z ddlZgdZddZdZdd Zdd Zdd Zd Zdd ZdZddZGddZddZdZdZdZy)a Miscellaneous Helpers for NetworkX. These are not imported into the base networkx namespace but can be accessed, for example, as >>> import networkx >>> networkx.utils.make_list_of_ints({1, 2, 3}) [1, 2, 3] >>> networkx.utils.arbitrary_element({5, 1, 7}) # doctest: +SKIP 1 N) defaultdictdeque)IterableIteratorSized)chaintee) flattenmake_list_of_intsdict_to_numpy_arrayarbitrary_elementpairwisegroupscreate_random_statecreate_py_random_statePythonRandomInterface nodes_equal edges_equal graphs_equalct|ttfrt|tr|S|g}|D]F}t|ttfrt|tr|j |;t ||Ht |S)z>Return flattened version of (possibly nested) iterable object.) isinstancerrstrappendr tuple)objresultitems ]/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/networkx/utils/misc.pyr r ,sr cHe, -C1E  ~"$5 12js6K MM$  D& ! " =ct|tsGg}|D]>}d|} t|}||k7rt j ||j |@|St|D]F\}}d|}t|tr t|}||k7rt j ||||<H|S#t$rt j |dwxYw#t$rt j |dwxYw)a*Return list of ints from sequence of integral numbers. All elements of the sequence must satisfy int(element) == element or a ValueError is raised. Sequence is iterated through once. If sequence is a list, the non-int values are replaced with ints. So, no new list is created zsequence is not all integers: N)rlistint ValueErrornx NetworkXErrorr enumerate)sequencerierrmsgiiindxs rr r :s h % A5aS9F 9VQw&&v.. MM"   X& a1!5 a   5QB 7""6* *  O% 9&&v.D8 9 5""6* 4 5s B. C. C C1c^ t||S#ttf$rt||cYSwxYw)zPConvert a dictionary of dictionaries to a numpy array with optional mapping.)_dict_to_numpy_array2AttributeError TypeError_dict_to_numpy_array1)dmappings rr r ^s71$Q00 I &1%Q001s ,,c ddl}|wt|j}|jD]$\}}|j |j&t t |tt|}t|}|j||f}|jD]+\}} |jD]\} } ||| || | f<-|S#t$rY%wxYw)zYConvert a dictionary of dictionaries to a 2d numpy array with optional mapping. rN) numpysetkeysitemsupdatedictziprangelenzerosKeyError) r1r2npskvnak1r(k2js rr-r-is  MGGI DAq HHQVVX  s1eCFm,- G A !QAA]]_ EB B%)!Q$  H  s C C('C(c ddl}|@t|j}tt |t t |}t |}|j|}|jD]\}}||}||||<|S)zJConvert a dictionary of numbers to a 1d numpy array with optional mapping.rN) r4r5r6r9r:r;r<r=r7)r1r2r?r@rCrDrEr(s rr0r0s~ Ms1eCFm,- G A  AA BKu! Hrc`t|tr tdtt |S)aReturns an arbitrary element of `iterable` without removing it. This is most useful for "peeking" at an arbitrary element of a set, but can be used for any list, dictionary, etc., as well. Parameters ---------- iterable : `abc.collections.Iterable` instance Any object that implements ``__iter__``, e.g. set, dict, list, tuple, etc. Returns ------- The object that results from ``next(iter(iterable))`` Raises ------ ValueError If `iterable` is an iterator (because the current implementation of this function would consume an element from the iterator). Examples -------- Arbitrary elements from common Iterable objects: >>> nx.utils.arbitrary_element([1, 2, 3]) # list 1 >>> nx.utils.arbitrary_element((1, 2, 3)) # tuple 1 >>> nx.utils.arbitrary_element({1, 2, 3}) # set 1 >>> d = {k: v for k, v in zip([1, 2, 3], [3, 2, 1])} >>> nx.utils.arbitrary_element(d) # dict_keys 1 >>> nx.utils.arbitrary_element(d.values()) # dict values 3 `str` is also an Iterable: >>> nx.utils.arbitrary_element("hello") 'h' :exc:`ValueError` is raised if `iterable` is an iterator: >>> iterator = iter([1, 2, 3]) # Iterator, *not* Iterable >>> nx.utils.arbitrary_element(iterator) Traceback (most recent call last): ... ValueError: cannot return an arbitrary item from an iterator Notes ----- This function does not return a *random* element. If `iterable` is ordered, sequential calls will return the same value:: >>> l = [1, 2, 3] >>> nx.utils.arbitrary_element(l) 1 >>> nx.utils.arbitrary_element(l) 1 z0cannot return an arbitrary item from an iterator)rrr#nextiter)iterables rr r s*~(H%KLL X rct|\}}t|d}|durt|t||fSt||S)z&s -> (s0, s1), (s1, s2), (s2, s3), ...NT)r rJr:r)rLcyclicrDbfirsts rrrsC x=DAq DME ~1eAx()) q!9rctt}|jD]\}}||j|t |S)aConverts a many-to-one mapping into a one-to-many mapping. `many_to_one` must be a dictionary whose keys and values are all :term:`hashable`. The return value is a dictionary mapping values from `many_to_one` to sets of keys from `many_to_one` that have that value. Examples -------- >>> from networkx.utils import groups >>> many_to_one = {"a": 1, "b": 1, "c": 2, "d": 3, "e": 3} >>> groups(many_to_one) # doctest: +SKIP {1: {'a', 'b'}, 2: {'c'}, 3: {'e', 'd'}} )rr5r7addr9) many_to_one one_to_manyrBrAs rrrsG c"K!!#1A1  rchddl}|||jur |jjjSt ||jj r|St |t r|jj |St ||jjr|S|d}t|)aReturns a numpy.random.RandomState or numpy.random.Generator instance depending on input. Parameters ---------- random_state : int or NumPy RandomState or Generator instance, optional (default=None) If int, return a numpy.random.RandomState instance set with seed=int. if `numpy.random.RandomState` instance, return it. if `numpy.random.Generator` instance, return it. if None or numpy.random, return the global random number generator used by numpy.random. rNzW cannot be used to create a numpy.random.RandomState or numpy.random.Generator instance) r4randommtrand_randr RandomStater" Generatorr#) random_stater?msgs rrrs|ryy8yy%%%, 5 56,$yy$$\22, 3 34 .* * S/rcReZdZd dZdZdZd dZdZdZdZ d Z d Z d Z d Z y)rNc ddl}|&j j j|_y||_y#t$rd}tj|tYWwxYw)Nrz.numpy not found, only random.random available.) r4 ImportErrorwarningswarn ImportWarningrVrWrX_rng)selfrngr?r\s r__init__zPythonRandomInterface.__init__sS .  ; ((..DIDI .BC MM#} - .s6%AAc6|jjSNrcrV)rds rrVzPythonRandomInterface.randomsyy!!rcH|||z |jjzzSrhri)rdrDrOs runiformzPythonRandomInterface.uniform"s#AETYY--////rcddl}t|j|jjr|jj ||S|jj ||SNrr4rrcrVrZintegersrandintrdrDrOr?s r randrangezPythonRandomInterface.randrange%sK dii!4!4 599%%a+ +yy  A&&rcddl}t|j|jjr*|jj dt |}||S|jjdt |}||Srm)r4rrcrVrZror<rp)rdseqr?idxs rchoicezPythonRandomInterface.choice.sg dii!4!4 5))$$QC1C3x))##As3x0C3xrc:|jj||Srh)rcnormal)rdmusigmas rgausszPythonRandomInterface.gauss7syyE**rc8|jj|Srh)rcshuffle)rdrts rr}zPythonRandomInterface.shuffle:syy  %%rcR|jjt||fdS)NF)sizereplace)rcrvr!)rdrtrAs rsamplezPythonRandomInterface.sample@s$yyS eDDrcddl}t|j|jjr|jj ||dzS|jj ||dzS)Nrrnrqs rrpzPythonRandomInterface.randintCsS dii!4!4 599%%aQ/ /yy  AE**rc>|jjd|z S)Nr)rc exponential)rdscales r expovariatez!PythonRandomInterface.expovariateKsyy$$QY//rc8|jj|Srh)rcpareto)rdshapes r paretovariatez#PythonRandomInterface.paretovariateOsyy&&rrh)__name__ __module__ __qualname__rfrVrkrrrvr{r}rrprrrrrrs: "0'+& E+0'rrcddl} ddl}||jur)t|jjjSt ||jj |jjfr t|St |tr|S |||ur |jSt ||jr|St |tr|j|S|d}t|#t$rYfwxYw)aReturns a random.Random instance depending on input. Parameters ---------- random_state : int or random number generator or None (default=None) If int, return a random.Random instance set with seed=int. if random.Random instance, return it. if None or the `random` package, return the global random number generator used by `random`. if np.random package, return the global numpy random number generator wrapped in a PythonRandomInterface class. if np.random.RandomState or np.random.Generator instance, return it wrapped in PythonRandomInterface if a PythonRandomInterface instance, return it rNz4 cannot be used to generate a random.Random instance) rVr4rrWrXrrYrZr__instRandomr"r#)r[rVr?r\s rrr^s    299 $()9)9)?)?@ @ lRYY%:%:BII>$':+;+;D+AA  !  ! ! rc|j|jk(xr4|j|jk(xr|j|jk(S)aCheck if graphs are equal. Equality here means equal as Python objects (not isomorphism). Node, edge and graph data must match. Parameters ---------- graph1, graph2 : graph Returns ------- bool True if graphs are equal, False otherwise. )adjnodesgraph)graph1graph2s rrrsC  fjj  ) LLFLL ( ) LLFLL (rrh)F)__doc__sysuuidr`rrrcollections.abcrrr itertoolsrr networkxr$__all__r r r r-r0r rrrrrrrrrrrrs|  *55  * !H1 .  B L,<>'>'X%P64nr