ó ʽ÷Xc@`sOddlmZmZmZddlZddlZddlZddlZddlZddl m Z ddl Z ddl Z e jdd„Zddd„Zdd„Zeddd„Zd „Zd efd „ƒYZd „Zeeeeed „Zy/e je dddddgƒZd„ZWnek rJd„ZnXdS(i(tdivisiontprint_functiontabsolute_importN(t namedtuplecC`s_tj|dtƒ|}|dk r7|j|ƒ}nt|tjƒs[tj|ƒ}n|S(s"Return an array of all value. tdtypeN(tnptonestbooltNonetastypet isinstancetndarraytasarray(tshapetvaluettypecodetout((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt _valarrays  c `s(|dkr3|dkr'tdƒ‚qNtj}n|dk rNtdƒ‚ntj|Œ}t‡fd†|Dƒƒ}tjg|D]}|jj^q†ƒ}t tj |dƒd|d|ƒ}tj |ˆ||Œƒ|dk r$t‡fd†|Dƒƒ}tj |ˆ||Œƒn|S( s¼ np.where(cond, x, fillvalue) always evaluates x even where cond is False. This one only evaluates f(arr1[cond], arr2[cond], ...). For example, >>> a, b = np.array([1, 2, 3, 4]), np.array([5, 6, 7, 8]) >>> def f(a, b): return a*b >>> _lazywhere(a > 2, (a, b), f, np.nan) array([ nan, nan, 21., 32.]) Notice it assumes that all `arrays` are of the same shape, or can be broadcasted together. s%One of (fillvalue, f2) must be given.s)Only one of (fillvalue, f2) can be given.c3`s!|]}tjˆ|ƒVqdS(N(Rtextract(t.0tarr(tcond(s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pys 3siRRc3`s"|]}tjˆ|ƒVqdS(N(RR(RR(R(s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pys 8sN( Rt ValueErrorRtnantbroadcast_arraysttuplet mintypecodeRtcharRR tplace( Rtarraystft fillvaluetf2ttemptattcodeR((Rs//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt _lazywheres    (% c `sþtj|Œ}tjg|D]}|jj^qƒ}ttj|dƒd|d|ƒ}x›tt|ƒƒD]‡}||||}‰tj ˆt kƒr¥qontjˆ|dƒ\‰} t ‡fd†|Dƒƒ} tj |ˆ|| ŒƒqoW|S(s? Mimic `np.select(condlist, choicelist)`. Notice it assumes that all `arrays` are of the same shape, or can be broadcasted together. All functions in `choicelist` must accept array arguments in the order given in `arrays` and must return an array of the same shape as broadcasted `arrays`. Examples -------- >>> x = np.arange(6) >>> np.select([x <3, x > 3], [x**2, x**3], default=0) array([ 0, 1, 4, 0, 64, 125]) >>> _lazyselect([x < 3, x > 3], [lambda x: x**2, lambda x: x**3], (x,)) array([ 0., 1., 4., 0., 64., 125.]) >>> a = -np.ones_like(x) >>> _lazyselect([x < 3, x > 3], ... [lambda x, a: x**2, lambda x, a: a * x**3], ... (x, a), default=np.nan) array([ 0., 1., 4., nan, -64., -125.]) iRRc3`s!|]}tjˆ|ƒVqdS(N(RR(RR(R(s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pys as( RRRRRRR trangetlentalltFalseRR( tcondlistt choicelistRtdefaultR"R#Rtindextfunct_R!((Rs//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt _lazyselect>s(%tCcC`sîtj|ƒ}|dkr'|j}nt|dƒsB|f}ntjtj|ƒ|j }tj ||dtj ƒ}|j dd|}|dkr©||}n||||d!d }tj |||d|ƒ}|jdƒ|S(sôAllocate a new ndarray with aligned memory. Primary use case for this currently is working around a f2py issue in Numpy 1.9.1, where dtype.alignment is such that np.zeros() does not necessarily create arrays aligned up to it. t__len__itdataiiÿÿÿÿtorderN(RRRt alignmentthasattrt functoolstreducetoperatortmultitemsizetemptytuint8t__array_interface__R tfill(R RR3taligntsizetbuftoffsetR2((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt_aligned_zerosfs      cC`s6|jdk r2|j|jjdkr2|jƒS|S(sÄReturn an array equivalent to the input array. If the input array is a view of a much larger array, copy its contents to a newly allocated array. Otherwise, return the input unchaged. iN(tbaseRR@tcopy(tarray((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt _prune_array€s( tDeprecatedImportcB`s)eZdZd„Zd„Zd„ZRS(s2 Deprecated import, with redirection + warning. Examples -------- Suppose you previously had in some module:: from foo import spam If this has to be deprecated, do:: spam = DeprecatedImport("foo.spam", "baz") to redirect users to use "baz" module instead. cC`s6||_||_t|jƒtj|j|_dS(N(t _old_namet _new_namet __import__tsystmodulest_mod(tselftold_module_nametnew_module_name((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt__init__œs   cC`s t|jƒS(N(tdirRN(RO((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt__dir__¢scC`s0tjd|j|jftƒt|j|ƒS(Ns'Module %s is deprecated, use %s instead(twarningstwarnRIRJtDeprecationWarningtgetattrRN(ROtname((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt __getattr__¥s (t__name__t __module__t__doc__RRRTRZ(((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyRHŠs  cC`s€|dks|tjkr(tjjjSt|tjtjfƒrStjj |ƒSt|tjj ƒrl|St d|ƒ‚dS(s6Turn seed into a np.random.RandomState instance If seed is None (or np.random), return the RandomState singleton used by np.random. If seed is an int, return a new RandomState instance seeded with seed. If seed is already a RandomState instance, return it. Otherwise raise ValueError. s=%r cannot be used to seed a numpy.random.RandomState instanceN( RRtrandomtmtrandt_randR tnumberstIntegraltintegert RandomStateR(tseed((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pytcheck_random_state­s  c C`s÷|s<ddl}|jj|ƒr<d}t|ƒ‚q<n|sftjj|ƒrftdƒ‚qfn|rutjntj}||ƒ}|sº|j tj dƒkrºtdƒ‚qºn|rótj |j tj ƒsó||dtj ƒ}qón|S(sB Helper function for scipy argument validation. Many scipy linear algebra functions do support arbitrary array-like input arguments. Examples of commonly unsupported inputs include matrices containing inf/nan, sparse matrix representations, and matrices with complicated elements. Parameters ---------- a : array_like The array-like input. check_finite : bool, optional Whether to check that the input matrices contain only finite numbers. Disabling may give a performance gain, but may result in problems (crashes, non-termination) if the inputs do contain infinities or NaNs. Default: True sparse_ok : bool, optional True if scipy sparse matrices are allowed. objects_ok : bool, optional True if arrays with dype('O') are allowed. mask_ok : bool, optional True if masked arrays are allowed. as_inexact : bool, optional True to convert the input array to a np.inexact dtype. Returns ------- ret : ndarray The converted validated array. iNsxSparse matrices are not supported by this function. Perhaps one of the scipy.sparse.linalg functions would work instead.smasked arrays are not supportedtOsobject arrays are not supportedR( t scipy.sparsetsparsetissparseRRtmat isMaskedArraytasarray_chkfiniteR Rt issubdtypetinexacttfloat_( R"t check_finitet sparse_okt objects_oktmask_okt as_inexacttscipytmsgttoarray((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyt_asarray_validatedÀs"#  tArgSpectargstvarargstkeywordstdefaultscC`sNtj|ƒ}g|jjƒD]$}|jtjjkr|j^q}g|jjƒD]$}|jtjjkrY|j^qY}|r“|dnd}g|jjƒD]$}|jtjj kr©|j^q©}|rã|dnd}g|jjƒD]6}|jtjjkrù|j |j k rù|j ^qùp8d}t ||||ƒS(söinspect.getargspec replacement using inspect.signature. inspect.getargspec is deprecated in python 3. This is a replacement based on the (new in python 3.3) `inspect.signature`. Parameters ---------- func : callable A callable to inspect Returns ------- argspec : ArgSpec(args, varargs, varkw, defaults) This is similar to the result of inspect.getargspec(func) under python 2.x. NOTE: if the first argument of `func` is self, it is *not*, I repeat *not* included in argspec.args. This is done for consistency between inspect.getargspec() under python 2.x, and inspect.signature() under python 3.x. iN(tinspectt signaturet parameterstvaluestkindt ParametertPOSITIONAL_OR_KEYWORDRYtVAR_POSITIONALRt VAR_KEYWORDR+R;Rz(R-tsigtpR{R|tvarkwR~((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pytgetargspec_no_self s$$$!cC`s9tj|ƒ}|jddkr5|jjdƒn|S(sDinspect.getargspec replacement for compatibility with python 3.x. inspect.getargspec is deprecated in python 3. This wraps it, and *removes* `self` from the argument list of `func`, if present. This is done for forward compatibility with python 3. Parameters ---------- func : callable A callable to inspect Returns ------- argspec : ArgSpec(args, varargs, varkw, defaults) This is similar to the result of inspect.getargspec(func) under python 2.x. NOTE: if the first argument of `func` is self, it is *not*, I repeat *not* included in argspec.args. This is done for consistency between inspect.getargspec() under python 2.x, and inspect.signature() under python 3.x. iRO(Rt getargspecR{tpop(R-targspec((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyR‹:s( t __future__RRRR6R8RLRURat collectionsRRtnumpyRRRRR$R/tfloatRCRGtobjectRHRftTrueR(RyR€RzR‹tAttributeError(((s//tmp/pip-build-7oUkmx/scipy/scipy/_lib/_util.pyts0        $ ( #   C +