;>"^" @sJddlmZmZmZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl mZddlmZmZmZddlZddlmZmZddd d d d d dddddg Zdd ZddZGdddeZddZdd ZddZddZ dddZ!dd Z"da#da$d!d"d#Z%ej&d$d%Z'ed!dd&ej&d!d'd Z(ed!ej&d(dZ)ia*ej+d)ej,Z-ed!dd*d+dd,dZ.d-d.Z/d/d0Z0d1dZ1d2d3Z2dS)4)divisionabsolute_importprint_functionN) issubclass_ issubsctype issubdtype) set_module)ndarrayufuncasarray) getargspec formatargspecrrr deprecatedeprecate_with_doc get_includeinfosourcewholookfor byte_bounds safe_evalcCs|ddl}|jdkrEtjjtjj|jdd}n3ddlj}tjjtjj|jd}|S)a Return the directory that contains the NumPy \*.h header files. Extension modules that need to compile against NumPy should use this function to locate the appropriate include directory. Notes ----- When using ``distutils``, for example in ``setup.py``. :: import numpy as np ... Extension('extension_name', ... include_dirs=[np.get_include()]) ... rNcoreinclude) numpyZ show_configospathjoindirname__file__ numpy.corer)rdrr!A/home/birch/.local/lib/python3.5/site-packages/numpy/lib/utils.pyrs  *$cCs ||_|S)N)__name__)funcnamer!r!r"_set_function_name5s r&c@s7eZdZdZdddddZddZdS) _Deprecatez Decorator class to deprecate old functions. Refer to `deprecate` for details. See Also -------- deprecate NcCs||_||_||_dS)N)old_namenew_namemessage)selfr(r)r*r!r!r"__init__Fs  z_Deprecate.__init__cs|j}|j}|j}|dkrRy j}Wntk rQj}YnX|dkrkd|nd||f|dk rd|7fdd}t||}j}|dkr}n|jjd} t | dd} | dj r#| d |}nft | dd} x?| ddD]-} t | | kraP| t | d7} qHW|| d}d j |g}||_y j } Wntk rYnX|j j| |S) z: Decorator call. Refer to ``decorate``. Nz`%s` is deprecated!z%`%s` is deprecated, use `%s` instead! cs#tjtdd||S)z1`arrayrange` is deprecated, use `arange` instead! stacklevel)warningswarnDeprecationWarning)argskwds)depdocr$r!r"newfuncbsz$_Deprecate.__call__..newfuncr z )r(r)r*r#AttributeErrorr&__doc__ expandtabssplit _get_indentlstriplenr__dict__update)r+r$r3kwargsr(r)r*r6doclinesindentskipliner r!)r5r$r"__call__KsJ                z_Deprecate.__call__)r# __module__ __qualname__r:r,rHr!r!r!r"r':s r'cCsdtj}x?|D]7}t|j}|rt|t||}qW|tjkr`d}|S)zU Determines the leading whitespace that could be removed from all the lines. r)sysmaxsizer?r>min)rDrErGcontentr!r!r"r=s  r=cOsD|r3|d}|dd}t|||St||SdS)a Issues a DeprecationWarning, adds warning to `old_name`'s docstring, rebinds ``old_name.__name__`` and returns the new function object. This function may also be used as a decorator. Parameters ---------- func : function The function to be deprecated. old_name : str, optional The name of the function to be deprecated. Default is None, in which case the name of `func` is used. new_name : str, optional The new name for the function. Default is None, in which case the deprecation message is that `old_name` is deprecated. If given, the deprecation message is that `old_name` is deprecated and `new_name` should be used instead. message : str, optional Additional explanation of the deprecation. Displayed in the docstring after the warning. Returns ------- old_func : function The deprecated function. Examples -------- Note that ``olduint`` returns a value after printing Deprecation Warning: >>> olduint = np.deprecate(np.uint) DeprecationWarning: `uint64` is deprecated! # may vary >>> olduint(6) 6 rr7N)r')r3rBfnr!r!r"rs , cCs td|S)Nr*)r')msgr!r!r"srQc Cs|j}|dd}|d}|d}t|jj}|}}|dkrg||j|7}n]xPt||D]?\}} | dkr||d| 7}qw||d| 7}qwW||7}||fS)a# Returns pointers to the end-points of an array. Parameters ---------- a : ndarray Input array. It must conform to the Python-side of the array interface. Returns ------- (low, high) : tuple of 2 integers The first integer is the first byte of the array, the second integer is just past the last byte of the array. If `a` is not contiguous it will not use every byte between the (`low`, `high`) values. Examples -------- >>> I = np.eye(2, dtype='f'); I.dtype dtype('float32') >>> low, high = np.byte_bounds(I) >>> high - low == I.size*I.itemsize True >>> I = np.eye(2); I.dtype dtype('float64') >>> low, high = np.byte_bounds(I) >>> high - low == I.size*I.itemsize True datarstridesshapeNr7)Z__array_interface__r dtypeitemsizesizezip) aaiZa_dataZastridesZashapeZbytes_aZa_lowZa_highrTZstrider!r!r"rs       c Cs|dkr$tjj}|j}g}i}x|jD]}t||tr=||}t|}||jkr|d||}d}n|||<|}d}djt t |j } t |j } |j || | |jj|gq=Wd} d} d} d}xtt|D]}||}| t|dkr`t|d} | t|dkrt|d} | t|dkrt|d} |dr*|t|d7}q*Wt|dkrPtd| }td| }td| }d |d |d |d f}t|d d t|d d xtt|D]}||}td|dd |t|dd|dd |t|dd |dd |t|dd |dfqcWtd|dS)a% Print the NumPy arrays in the given dictionary. If there is no dictionary passed in or `vardict` is None then returns NumPy arrays in the globals() dictionary (all NumPy arrays in the namespace). Parameters ---------- vardict : dict, optional A dictionary possibly containing ndarrays. Default is globals(). Returns ------- out : None Returns 'None'. Notes ----- Prints out the name, shape, bytes and type of all of the ndarrays present in `vardict`. Examples -------- >>> a = np.arange(10) >>> b = np.ones(20) >>> np.who() Name Shape Bytes Type =========================================================== a 10 80 int64 b 20 160 float64 Upper bound on total bytes = 240 >>> d = {'x': np.arange(2.0), 'y': np.arange(3.0), 'txt': 'Some str', ... 'idx':5} >>> np.who(d) Name Shape Bytes Type =========================================================== x 2 16 float64 y 3 24 float64 Upper bound on total bytes = 40 Nz (%s)rr7z x r/ zName %s Shape %s Bytes %s Typer8r-=z%s %s %s %s %s %s %sz' Upper bound on total bytes = %d)rK _getframef_back f_globalskeys isinstancer idrmapstrrTnbytesappendrUr%ranger?intmaxprint)vardictframeZstacacher%varZidvnamestroriginalZshapestrZbytestrZmaxnameZmaxshapeZmaxbyteZ totalbyteskvalZsp1Zsp2Zsp3Zprvalr!r!r"rs^,        $ &  c Cst|}|}|}d}|j|}x|D]}||krOd} n|} |t|t| }||kr|dt|}|dd|d|}q4|| |}q4W|S)Nz, r7z, r8r/)r?r<) r% argumentswidthZ firstwidthrtnewstrZsepstrarglistargumentZaddstrr!r!r" _split_lineps     r|rcCst|ttg}|j|ji}|jg}|jg}xt|dkr[P|jd}xt|jD]f}t||t j rw||j}||krw||j}|j ||j ||||. __class__r#zclass: filezshape: z strides: z itemsize: z aligned: z contiguous: z fortran: zdata pointer: %s%sz byteorder: endr8|r]z%s%s%sF>z%sbig%sbigz %slittle%slittlez byteswap: ztype: %sN)rr])getattrtyperSrU byteorderrmrTrVflagsZaligned contiguousZfortranhexctypesZ_as_parameter_valuerK) objoutputextraZticZbpclsnmrSZendianbyteswapr!r!r"_infos8       rLc Csddl}ddl}t|ds6t|drB|j}nt|drZ|j}|dkrsttnt|trt|d|n{t|t rt dkrt |\a a d}g}xt D]}yt ||} t | |krtd|d|nE|jt | td |d|t| td |d||d 7}Wqtk ryYqXqW|dkrtd |d|qtd |d|nX|j|rK|j} tt|} t| | |kr t| | |} n | | } td| dd|t|j|d|n|j|rR|j} d} ywt|drtt|jj} | jd} t| d krd| d | d |D]u}|ddkrqt||d}|dk r1|j |j|p%d\}}td||fd|qWnt!j"ddkrt|t#j$rtd|j%jd|td|t|drtt|j&j} | jd} t| d krd| d | d >> np.info(np.polyval) # doctest: +SKIP polyval(p, x) Evaluate the polynomial p at x. ... When using a string for `object` it is possible to get multiple results. >>> np.info('fft') # doctest: +SKIP *** Found in numpy *** Core FFT routines ... *** Found in numpy.fft *** fft(a, n=None, axis=-1) ... *** Repeat reference found in numpy.fft.fftpack *** *** Total of 3 references found. *** rNZ_ppimport_importer_ppimport_module_ppimport_attrrz+ *** Repeat reference found in %s *** rz *** Found in %s ***-r7zHelp for %s not found.z+ *** Total of %d references found. ***r8r-z()r,z, (z Methods: _Nonez %s -- %sr_zInstance of class: rHr%z%szr:))pydocinspecthasattrrrrrdr rrg _namedictr _dictlistrermriKeyError isfunctionr#r r r?r|getdocisclassr,__func__r<r ExceptionZ allmethodsrZsplitdocrK version_inforZ InstanceTyperrHr%ismethod)objectZmaxwidthrZtoplevelrrZnumfoundZobjlistrrrr%rwargstrrzZdoc1methodsmethZthisobjZmethstrotherrCr!r!r"rs4                       !     c Csoddl}y:td|j|d|t|j|d|Wn"tk rjtdd|YnXdS)a Print or write to a file the source code for a NumPy object. The source code is only returned for objects written in Python. Many functions and classes are defined in C and will therefore not return useful information. Parameters ---------- object : numpy object Input object. This can be any object (function, class, module, ...). output : file object, optional If `output` not supplied then source code is printed to screen (sys.stdout). File object must be created with either write 'w' or append 'a' modes. See Also -------- lookfor, info Examples -------- >>> np.source(np.interp) #doctest: +SKIP In file: /usr/lib/python2.6/dist-packages/numpy/lib/function_base.py def interp(x, xp, fp, left=None, right=None): """.... (full docstring printed)""" if isinstance(x, (float, int, number)): return compiled_interp([x], xp, fp, left, right).item() else: return compiled_interp(x, xp, fp, left, right) The source code is only returned for objects written in Python. >>> np.source(np.array) #doctest: +SKIP Not available for this object. rNz In file: %s rzNot available for this object.)rrm getsourcefile getsourcer)rrrr!r!r"rs )  z[a-z0-9_]+\(.*[,=].*\)TFc soddl}t|||g}t|jjsFdSxjjD]\\}\}} } | dkrwqS|jtfddDrS|j|qSWddd dddddifd d fd d } |jd| ddj } | dt | g} x|dddD]}|\} }ddj jdD}y3|dj }t j |r|dj }Wntk rd}YnX| jd||fqHW|s| jd|dk r$|jdj | nGt | dkrX|j}|dj | ntdj | dS)a Do a keyword search on docstrings. A list of objects that matched the search is displayed, sorted by relevance. All given keywords need to be found in the docstring for it to be returned as a result, but the order does not matter. Parameters ---------- what : str String containing words to look for. module : str or list, optional Name of module(s) whose docstrings to go through. import_modules : bool, optional Whether to import sub-modules in packages. Default is True. regenerate : bool, optional Whether to re-generate the docstring cache. Default is False. output : file-like, optional File-like object to write the output to. If omitted, use a pager. See Also -------- source, info Notes ----- Relevance is determined only roughly, by checking if the keywords occur in the function name, at the start of a docstring, etc. Examples -------- >>> np.lookfor('binary representation') # doctest: +SKIP Search results for 'binary representation' ------------------------------------------ numpy.binary_repr Return the binary representation of the input number as a string. numpy.core.setup_common.long_double_representation Given a binary dump as given by GNU od -b, look for long double numpy.base_repr Return a string representation of a number in the given base system. ... rNrrc3s|]}|kVqdS)Nr!).0w)rCr!r" szlookfor..r$iclasscsd}dj|jjjddd|tfddD7}|tfddD7}|t d7}|j|d 7}|jd  d 7}|t| d d 7}|S)Nrr-r_cs"g|]}|krdqS)r!)rr) first_docr!r" &s z.lookfor..relevance..cs"g|]}|krdqS)r!)rr)r%r!r"r(s r^i.r\dii) rlowerstripr<sumr?getcountrl)r%Zdocstrkindindexr)kind_relevancewhats)rr%r" relevance"s.##zlookfor..relevancecs||S)Nr!)rY)rprr!r"relevance_value2sz lookfor..relevance_valuekeyzSearch results for '%s'r8rr7cSs(g|]}|jr|jqSr!)r)rrGr!r!r"r<s zlookfor..r-rvz %s %szNothing found.r\)zmodulezobjectii)r_lookfor_generate_cachergrr<itemsallrisortrr?r_function_signature_research IndexErrorwriteZgetpagerrm)whatrimport_modules regeneraterrfoundr% docstringrrrs help_textZixZdoclinesrZpagerr!)rprCrrrr"rsJ/ "   "     c#Csddl}tjddkr2ddlm}nddlm}|dkrTd}t|tryt|Wntk riSYnXtj |}nRt|t st|t ri}x'|D]}|j t |||qW|St|tkr| rtt|Si}|tt|.)r getmembersrdir)rrmembersr!)rr"rs   $rcCsddl}|j|S)a Protected string evaluation. Evaluate a string containing a Python literal expression without allowing the execution of arbitrary non-literal code. Parameters ---------- source : str The string to evaluate. Returns ------- obj : object The result of evaluating `source`. Raises ------ SyntaxError If the code has invalid Python syntax, or if it contains non-literal code. Examples -------- >>> np.safe_eval('1') 1 >>> np.safe_eval('[1, 2, 3]') [1, 2, 3] >>> np.safe_eval('{"foo": ("bar", 10.0)}') {'foo': ('bar', 10.0)} >>> np.safe_eval('import os') Traceback (most recent call last): ... SyntaxError: invalid syntax >>> np.safe_eval('open("/home/user/.ssh/id_dsa").read()') Traceback (most recent call last): ... ValueError: malformed node or string: <_ast.Call object at 0x...> rN)ast literal_eval)rrr!r!r"rs, cCs|jdkr|Stj||d}tj|d }tjj|r\|jd}|jdkr|dkr|dk r|jj tj |d<|}q|jj tj }n(tj |j dkrtj ||<|S) aj Utility function to check median result from data for NaN values at the end and return NaN in that case. Input result can also be a MaskedArray. Parameters ---------- data : array Input data to median function result : Array or MaskedArray Result of median function axis : {int, sequence of int, None}, optional Axis or axes along which the median was computed. out : ndarray, optional Output array in which to place the result. Returns ------- median : scalar or ndarray Median or NaN in axes which contained NaN in the input. rr7.FTNrr).r) rWnpZmoveaxisisnanmaZ isMaskedArrayZfilledndimrUrnanZ count_nonzeroZravel)rRresultZaxisoutrr!r!r"_median_nanchecks    r )3 __future__rrrrrKrrer0Znumpy.core.numerictypesrrrZnumpy.core.overridesrrr r r rr numpy.compatr r rrr&rr'r=rrrrr|rrrrrrrrcompileIrrrrrr r!r!r!r"sN         K  4  9i 3  3  z 0