3 dW.@s dZddlmZddlmZddlZddlZddlZddlmZm Z ddl Z ddZ dd Z d d Zd1d d ZddZddZddZddZddZd2ddZd3ddZd4ddZddZd d!Zd"d#Zd5d%d&Zd'd(Zd)d*Zd6d+d,ZGd-d.d.Zd7d/d0Z dS)8a Miscellaneous Helpers for NetworkX. These are not imported into the base networkx namespace but can be accessed, for example, as >>> import networkx >>> networkx.utils.is_list_of_ints([1, 2, 3]) True >>> networkx.utils.is_list_of_ints([1, 2, "spam"]) False ) defaultdict)dequeN)teechaincCsd}tj|tt|tS)zCheck if obj is string.zYis_string_like is deprecated and will be removed in 3.0.Use isinstance(obj, str) instead.)warningswarnDeprecationWarning isinstancestr)objmsgr 7/tmp/pip-build-7vycvbft/networkx/networkx/utils/misc.pyis_string_likes rc Cs*t|drdSy t|Wn dSdS)z: Return True if obj is iterable with a well-defined len().__iter__TF)hasattrlen)r r r riterable&s  rccsfEdHdS)z$ Return a generator with no members Nr r r r rempty_generator1srcCsbt| st|r|S|dkr"g}x4|D],}t| s>t|rJ|j|q(t||q(W|j|S)z@ Return flattened version of (possibly nested) iterable object. N)rrappendflatten __class__)r resultitemr r rr6s  rc Cst|tstg}x`|D]X}d|}y t|}Wn tk rNtj|dYnX||krbtj||j|qW|Sxrt|D]f\}}d|}t|trq~y t|}Wn tk rtj|dYnX||krtj||||<q~W|S)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)r listint ValueErrornxZ NetworkXErrorr enumerate)sequencerierrmsgiiZindxr r rmake_list_of_intsDs0          r#cCs.t|tsdSx|D]}t|tsdSqWdS)z( Return True if list is a list of ints. FT)r rr)Zintlistr r r ris_list_of_intshs    r$cCsd}tj|tt|S)z'Returns the string representation of t.zCmake_str is deprecated and will be removed in 3.0. Use str instead.)rrrr )xr r r rmake_strrs r&cCs ttjS)z Generate a unique node label.)r uuiduuid1r r r rgenerate_unique_nodeysr)cCsDddlm}dgdgdgddddgd }|tj|g}||d S) zOpens `filename` using system's default program. Parameters ---------- filename : str The path of the file to be opened. r)callopenzxdg-openzcmd.exez/Cstart)darwinlinuxZlinux2win32N) subprocessr*sysplatform)filenamer*Zcmdscmdr r rdefault_opener~s r6c Cs.y t||Sttfk r(t||SXdS)zPConvert a dictionary of dictionaries to a numpy array with optional mapping.N)dict_to_numpy_array2AttributeError TypeErrordict_to_numpy_array1)dmappingr r rdict_to_numpy_arrays r=c Csddl}|dkrVt|j}x"|jD]\}}|j|jq&Wtt|tt|}t|}|j ||f}xX|jD]L\}} xB|jD]6\} } y||| || | f<Wqt k rYqXqWqvW|S)zYConvert a dictionary of dictionaries to a 2d numpy array with optional mapping. rN) numpysetkeysitemsupdatedictziprangerzerosKeyError) r;r<r>skvnak1r Zk2jr r rr7s r7cCsrddl}|dkr2t|j}tt|tt|}t|}|j|}x(|jD]\}}||}||||<qNW|S)zTConvert a dictionary of numbers to a 1d numpy array with optional mapping. rN) r>r?r@rCrDrErrFrA)r;r<r>rHrKrLrMr r r rr:s  r:cCs$t|dpt|d}t||ko"|S)zMReturns True if and only if the given object is an iterator object. __next__next)riter)r Z has_next_attrr r r is_iteratorsrRcCst|rtdtt|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:: >>> arbitrary_element({3, 2, 1}) 1 >>> arbitrary_element("hello") 'h' This function raises a :exc:`ValueError` if `iterable` is an iterator (because the current implementation of this function would consume an element from the iterator):: >>> iterator = iter([1, 2, 3]) >>> arbitrary_element(iterator) Traceback (most recent call last): ... ValueError: cannot return an arbitrary item from an iterator z0cannot return an arbitrary item from an iterator)rRrrPrQ)rr r rarbitrary_elementsrScCst|dddS)zConsume the iterator entirely.r)maxlenN)r)iteratorr r rconsumesrVFcCs:t|\}}t|d}|dkr0t|t||fSt||S)z&s -> (s0, s1), (s1, s2), (s2, s3), ...NT)rrPrDr)rZcyclicrLbfirstr r rpairwises   rYcCs4tt}x"|jD]\}}||j|qWt|S)a Converts 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. For example:: >>> 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: {'d', 'e'}} )rr?rAaddrC)Z many_to_oneZ one_to_manyrJrIr r rgroupssr[cCs t|ttfs|Sttt|S)zConverts lists to tuples. For example:: >>> from networkx.utils import to_tuple >>> a_list = [1, 2, [1, 4]] >>> to_tuple(a_list) (1, 2, (1, 4)) )r tuplermapto_tuple)r%r r rr^s r^cCsbddl}|dks||jkr$|jjjSt||jjr6|St|trL|jj|S|d}t|dS)aReturns a numpy.random.RandomState instance depending on input. Parameters ---------- random_state : int or RandomState instance or None optional (default=None) If int, return a numpy.random.RandomState instance set with seed=int. if numpy.random.RandomState instance, return it. if None or numpy.random, return the global random number generator used by numpy.random. rNz? cannot be used to generate a numpy.random.RandomState instance)r>randommtrand_randr RandomStaterr) random_statenpr r r rcreate_random_state$s     rec @seZdZydddZWn$ek r:dZejeeYnXddZ ddZ dd d Z d d Z d dZ ddZddZddZddZddZdS)PythonRandomInterfaceNcCs&ddl}|dkr|jjj|_||_dS)Nr)r>r_r`ra_rng)selfrngr>r r r__init__@s zPythonRandomInterface.__init__z.numpy not found, only random.random available.cCs |jjS)N)rg random_sample)rhr r rr_KszPythonRandomInterface.randomcCs||||jjS)N)rgrk)rhrLrWr r runiformNszPythonRandomInterface.uniformcCs|jj||S)N)rgrandint)rhrLrWr r r randrangeQszPythonRandomInterface.randrangecCs||jjdt|S)Nr)rgrmr)rhseqr r rchoiceTszPythonRandomInterface.choicecCs|jj||S)N)rgZnormal)rhmusigmar r rgaussWszPythonRandomInterface.gausscCs |jj|S)N)rgshuffle)rhror r rrtZszPythonRandomInterface.shufflecCs|jjt||fddS)NF)sizereplace)rgrpr)rhrorIr r rsample`szPythonRandomInterface.samplecCs|jj||dS)N)rgrm)rhrLrWr r rrmcszPythonRandomInterface.randintcCs|jjd|S)Nrx)rgZ exponential)rhZscaler r r expovariategsz!PythonRandomInterface.expovariatecCs |jj|S)N)rgZpareto)rhshaper r r paretovariateksz#PythonRandomInterface.paretovariate)N)N)__name__ __module__ __qualname__rj ImportErrorr rr ImportWarningr_rlrnrprsrtrwrmryr{r r r rrf=s rfc Csddl}yHddl}||jkr*t|jjjSt||jjr@t|St|trN|SWntk rdYnX|dksv||kr||jSt||j r|St|t r|j |S|d}t |dS)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 instance, return it wrapped in PythonRandomInterface if a PythonRandomInterface instance, return it rNz4 cannot be used to generate a random.Random instance) r_r>rfr`rar rbr_instRandomrr)rcr_rdr r r rcreate_py_random_statezs&      r)N)N)N)N)F)N)N)!__doc__ collectionsrrrr2r' itertoolsrrZnetworkxrrrrrr#r$r&r)r6r=r7r:rRrSrVrYr[r^rerfrr r r r s8     $      =