3
���hS � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z d dl
m Z d dl m
Z
mZmZ d dlZd dlmZmZ dddd d
d d d
ddddg Zdd � Zdd� ZG dd� d�Zdd� Zdd � Zdd� Zd d� Z d3d d�Z d d � Z da da d4d!d"�Z!ej"fd#d$�Z#e d �dd%ej"d fd&d ��Z$e d �ej"fd'd
��Z%i Z&ej'd(ej(�Z)e d �d5d+d��Z*d,d-� Z+d.d/� Z,d0d� Z-d1d2� Z.dS )6� N)� issubclass_� issubsctype�
issubdtype)�
set_module)�ndarray�ufunc�asarray)�
getargspec�
formatargspecr r r � deprecate�deprecate_with_doc� get_include�info�source�who�lookfor� byte_bounds� safe_evalc C sT ddl } | jdkr.tjjtjj| j�dd�}n"ddlj}tjjtjj|j�d�}|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()])
...
r N�core�include) �numpyZ show_config�os�path�join�dirname�__file__�
numpy.corer )r �dr � r �0/tmp/pip-build-5_djhm0z/numpy/numpy/lib/utils.pyr
s
c C s
|| _ | S )N)�__name__)�func�namer r r �_set_function_name4 s r# c @ s" e Zd ZdZddd�Zdd� ZdS )�
_Deprecatez�
Decorator class to deprecate old functions.
Refer to `deprecate` for details.
See Also
--------
deprecate
Nc C s || _ || _|| _d S )N)�old_name�new_name�message)�selfr% r& r' r r r �__init__E s z_Deprecate.__init__c s~ | j }| j}| j}|dkr@y
�j}W n tk
r> �j}Y nX |dkrRd| � n d||f � |dk rr� d| 7 � � �fdd�}t||�}�j}|dkr�� }n�|j� jd�} t | dd� �}
| d j
� r�|
d | }nRt | d �d } x4| dd� D ]$} t | �|
k�rP | t | �d 7 } q�W || d� }t j
� d |
�� d
j� |g�}||_y
�j}
W n tk
�rl Y nX |jj|
� |S ) z:
Decorator call. Refer to ``decorate``.
Nz`%s` is deprecated!z%`%s` is deprecated, use `%s` instead!�
c s t j� tdd� �| |�S )z1`arrayrange` is deprecated, use `arange` instead!� )�
stacklevel)�warnings�warn�DeprecationWarning)�args�kwds)�depdocr! r r �newfunca s z$_Deprecate.__call__.<locals>.newfunc� r � z
)r% r& r' r �AttributeErrorr# �__doc__�
expandtabs�split� _get_indent�lstrip�len�textwrap�indentr �__dict__�update)r( r! r0 �kwargsr% r& r' r3 �doc�linesr> �skip�liner r )r2 r! r �__call__J sL
z_Deprecate.__call__)NNN)r �
__module__� __qualname__r7 r) rF r r r r r$ 9 s
r$ c C sH t j}x.| D ]&}t|j� �}|r t|t|�| �}q W |t jkrDd}|S )zU
Determines the leading whitespace that could be removed from all the lines.
r )�sys�maxsizer< r; �min)rC r> rE �contentr r r r: � s
r: c O s4 | r&| d }| dd� } t | |�|�S t | |�S dS )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
r r4 N)r$ )r0 rA �fnr r r r � s
, c C s
t | d�S )N)r' )r$ )�msgr r r �<lambda>� s rO c
C s� | j }|d d }|d }|d }t| �jj}| }}|dkrN|| j| 7 }nJx@t||�D ]2\}} | dk r|||d | 7 }qZ||d | 7 }qZW ||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
�datar �strides�shapeNr4 )Z__array_interface__r �dtype�itemsize�size�zip)
�aZaiZa_dataZastridesZashapeZbytes_aZa_lowZa_highrR Zstrider r r r � s c
C sP | dkrt j� j}|j} g }i }x�| j� D ]�}t| | t�r*| | }t|�}||j� krn|d|| }d}n|||< |}d}djt t
|j ��} t
|j �}
|j
|| |
|jj|g� q*W d} d} d}
d}x�tt|��D ]|}|| }| t|d �k r�t|d �} | t|d �k �rt|d �} |
t|d �k �r8t|d �}
|d r�|t|d �7 }q�W t|�dk�r�td| �}td| �}td|
�}d |d
|d
|d
f }t|d d t|�d
d � x�tt|��D ]t}|| }td|d d
|t|d � d |d d
|t|d � d
|d d
|t|d � d
|d f � �q�W td| � 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)r r4 z x r+ � �
z Name %s Shape %s Bytes %s Typer5 r* �=� z%s %s %s %s %s %s %s� z'
Upper bound on total bytes = %d)rI � _getframe�f_back� f_globals�keys�
isinstancer �idr �map�strrR �nbytes�appendrS r" �ranger< �int�max�print)�vardict�frameZsta�cacher" �varZidv�namestr�originalZshapestrZbytestrZmaxnameZmaxshapeZmaxbyteZ
totalbytes�k�valZsp1Zsp2Zsp3Zprvalr r r r s^ ,
c
C s� t | �}|}| }d}|j|�}xp|D ]h}||kr6d} n|} |t |� t | � }||kr�|d t |� }|d d|d | }q$|| | }q$W |S )Nz, � r4 z,
r5 r+ )r< r9 )
r" � arguments�widthZ
firstwidthrq ZnewstrZsepstr�arglistZargumentZaddstrr r r � _split_linep s
rw r c C s� t | t� t� g �} | j| ji}| jg}| jg}xvt|�dkr>P |jd�}xX|j� D ]L}t|| t j
�rR|| j}||krR|| j}|j |� |j |� |||< qRW q0W ||fS )Nr ) �
__import__�globals�localsr r? r< �popr` ra �types�
ModuleTyperf )�moduleZthedictZdictlistZ
totraverseZthisdict�x�modnameZmoddictr r r �
_makenamedict� s"
r� c
C sx d}d}dd� }t | dt| ��}t |d|�}| j}| jj}td||d� td| j|d� td ||d� td
| j|d� td || jj �|d� td || jj
�|d� td
| jj |d� tdt | j
jj�|f |d� tdd|d� |d k�rtd|tj|f |d� d} nH|dk�r2td||f |d� tjdk} n td||f |d� tjdk} td|| �|d� td | j |d� d S ) aO Provide information about ndarray obj.
Parameters
----------
obj : ndarray
Must be ndarray, not checked.
output
Where printed output goes.
Notes
-----
Copied over from the numarray module prior to its removal.
Adapted somewhat as only numpy is an option now.
Called by info.
rs c S s | S )Nr )r r r r rO � s z_info.<locals>.<lambda>� __class__r zclass: )�filezshape: z strides: z
itemsize: z aligned: z contiguous: z fortran: zdata pointer: %s%sz byteorder: r5 )�endr� �|rZ z%s%s%sF�>z%sbig%s�bigz
%slittle%s�littlez
byteswap: ztype: %sN)r� rZ )�getattr�typerQ rS � byteorderrj rR rT �flagsZaligned�
contiguousZfortran�hex�ctypesZ_as_parameter_�valuerI )
�obj�output�extraZticZbp�cls�nmrQ ZendianZbyteswapr r r �_info� s8
r� �L c
C s* ddl }ddl}t| d�s$t| d�r,| j} nt| d�r<| j} | dkrPtt� �n�t| t�rjt| |d� �n�t| t ��rTt
dkr�t |�\a
a d}g }x�t D ]�}ylt
| | } t
| �|kr�td| |d� n6|jt
| �� td | |d� t| � td
| |d� |d 7 }W q� tk
�r Y q�X q�W |dk�r@td | |d� ntd
| |d� �n�|j| ��r�| j}
tt| �� } t|
| �|k�r�t|
| |�} n|
| } td| d |d� t|j| �|d� �nb|j| ��rX| j}
d} yZt| d��r2tt| jj�� } | jd�}
t|
�d k�r2d|
d |
d < dj |
d d� �} W n t k
�rJ Y nX t|
| �|k�rlt|
| |�} n|
| } td| d |d� |j| �}|dk�r�t| d��r�t|j| j�|d� nt|j| �|d� |j | �}|g k�r&td|d� xd|D ]\}|d dk�r �q�t | |d�}|dk �r<|j |j|��p4d�\}}td||f |d� �q�W n�|j!| ��r| j}
tt| j�� } | jd�}
t|
�d k�r�d|
d |
d < dj |
d d� �} nd} t|
| �|k�r�t|
| |�} n|
| } td| d |d� t|j| �|d� n t| d��r&t|j| �|d� dS )a�
Get help information for a function, class, or module.
Parameters
----------
object : object or str, optional
Input object or name to get information about. If `object` is a
numpy object, its docstring is given. If it is a string, available
modules are searched for matching objects. If None, information
about `info` itself is returned.
maxwidth : int, optional
Printing width.
output : file like object, optional
File like object that the output is written to, default is
``stdout``. The object has to be opened in 'w' or 'a' mode.
toplevel : str, optional
Start search at this level.
See Also
--------
source, lookfor
Notes
-----
When used interactively with an object, ``np.info(obj)`` is equivalent
to ``help(obj)`` on the Python prompt or ``obj?`` on the IPython
prompt.
Examples
--------
>>> 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. ***
r NZ_ppimport_importer�_ppimport_module�_ppimport_attr)r� z+
*** Repeat reference found in %s *** )r� z *** Found in %s ***�-r4 zHelp for %s not found.z+
*** Total of %d references found. ***r5 r* z()r) z, �(z
Methods:
�_�Nonez %s -- %sr7 )"�pydoc�inspect�hasattrr� r� r ra r r� rd � _namedictr� � _dictlistrb rj rf �KeyErrorZ
isfunctionr r
r r< rw �getdoc�isclassr) �__func__r9 r � ExceptionZ
allmethodsr� ZsplitdocZismethod)�objectZmaxwidthr� �toplevelr� r� ZnumfoundZobjlistro r� r" rt Zargstrrv Zdoc1�methods�methZthisobjZmethstr�otherr r r r � s� 4
c C sZ ddl }y,td|j| � |d� t|j| �|d� W n tk
rT td|d� Y nX dS )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.
r Nz In file: %s
)r� z Not available for this object.)r� rj Z
getsourcefileZ getsourcer� )r� r� r� r r r r z s )z[a-z0-9_]+\(.*[,=].*\)TFc
s� ddl }t|||�� g }t| �j� j� ��s0dS xL� j� D ]@\}\}} }
| dkrRq:|j� �t�fdd��D ��r:|j|� q:W ddddd����fd d
��� �fd d �} |j| d
� ddj �� } | dt
| � g}
x�|ddd � D ]�}� | \�} }dd� �j � jd�D �}y(|d j � }t j
|��r2|d j � }W n tk
�rN d}Y nX |
jd||f � q�W |�sv|
jd� |dk �r�|jdj |
�� n4t
|
�dk�r�|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.
...
r Nr~ r� c 3 s | ]}|� kV qd S )Nr )�.0�w)rB r r � <genexpr>� s zlookfor.<locals>.<genexpr>i� )r! �classr~ r� c s� d}dj |j� j� jd�d d� �� |t� fdd��D ��7 }|t�fdd��D ��7 }|t�� d 7 }|�j|d �7 }|�jd � d
7 }|t| d d
�7 }|S )Nr r* r\ c s g | ]}|� krd �qS )�� r )r� r� )� first_docr r �
<listcomp> s z.lookfor.<locals>.relevance.<locals>.<listcomp>c s g | ]}|� krd �qS )� r )r� r� )r" r r r� s r[ i� �.rY �d i���i����) r �lower�stripr9 �sumr< �get�countri )r" Zdocstr�kind�index�r)�kind_relevance�whats)r� r" r � relevance� s zlookfor.<locals>.relevancec s �| f� | �� S )Nr )rW )rm r� r r �relevance_value s z lookfor.<locals>.relevance_value)�keyzSearch results for '%s'r5 r� r4 c S s g | ]}|j � r|j � �qS r )r� )r� rE r r r r� s zlookfor.<locals>.<listcomp>r* rs z %s
%szNothing found.rY )r~ r� i���i������)r� �_lookfor_generate_cacherd r� r9 �items�allrf �sortr r< r� �_function_signature_re�search�
IndexError�writeZgetpagerrj )�whatr~ �import_modules�
regenerater� r� �foundr" Z docstringr� r� r� �s� help_text�ixZdoclinesr� Zpagerr )rm rB r� r� r� r r � sJ /
c # C s� ddl }ddlm} | dkr d} t| t�rXy t| � W n tk
rJ i S X tj| } n<t| t �slt| t
�r�i }x | D ]}|j t |||�� qvW |S t
| �tkr�| r�tt
| � S i }|tt
| �< i }d}| j| fg} �x�| �r�| jd�\}
} t
| �|k� r�q�d|t
| �< |d7 }d} |j| ��r2d} y
| j}
W n tk
�rJ d}
Y nX |�rRt| d ��rRx�| jD ]�}x�tj|�D ]�}tjj||�}tjj||d
�}tjj|��r�|jd ��r�|dd� }ntjj|��rv|}n�qv|d
k�r�qvyDtj }tj }z$|� t_ |� t_ td|
|f � W d|t_ |t_ X W n t k
�rD �wvY nX �qvW �qfW x�t | �D ]�\}}y$t |dd|
|f �}t |dd�}W n& t!k
�r� d|
|f }d}Y nX d|k�r�|�r�d||f }|j"|
d ��s�t|t#��r\n�q\n$|j|��p |
dk�p ||
k�s�q\| j$d|
|f |f� �q\W nP|j%| ��rrd} x>t | �D ] \}}| j$d|
|f |f� �qLW nt| d��r�d} y|j&| �}W n t!k
�r� d}Y nX |dk r�|| |f||
< q�W |S )a�
Generate docstring cache for given module.
Parameters
----------
module : str, None, module
Module for which to generate docstring cache
import_modules : bool
Whether to import sub-modules in packages.
regenerate : bool
Re-generate the docstring cache
Returns
-------
cache : dict {obj_full_name: (docstring, kind, index), ...}
Docstring cache for the module, either cached one (regenerate=False)
or newly generated.
r N)�StringIOr Tr4 r� r~ �__path__z __init__.pyz.pyr\ r) z%s.%sr rG r� r� rF r! ���)'r� �ior� ra rd rx � ImportErrorrI �modules�list�tupler@ r� rb �_lookfor_cachesr r{ Zismodule�__all__r6 r� r� r �listdirr r �isfile�endswith�stdout�stderr�
BaseException� _getmembersr� � NameError�
startswithr rf r� r� )r~ r� r� r� r� rm �mod�seenr� �stackr" �itemr� Z_all�pthZmod_pathZthis_pyZinit_pyZ to_importZ
old_stdoutZ
old_stderr�n�vZ item_name�mod_namerB r r r r� 1 s�
r� c sF dd l }y|j� �}W n* tk
r@ � fdd�t� �D �}Y nX |S )Nr c s$ g | ] }t � |�r|t� |�f�qS r )r� r� )r� r )r� r r r� � s z _getmembers.<locals>.<listcomp>)r� �
getmembersr� �dir)r� r� �membersr )r� r r� � s r� c C s ddl }|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...>
r N)�astZ literal_eval)r r� r r r r � s ,c C s� | j dkr|S tj| |d�} tj| d �}tjj|�r@|jd�}|jdkr�|dkr�|dk rr| jj tj
�|d< |}q�| jj tj
�}n tj |j � �dkr�tj
||< |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.
r r4 .FTNr� r� ).r� )
rU �npZmoveaxis�isnan�maZ
isMaskedArrayZfilled�ndimrS r� �nanZ
count_nonzeroZravel)rP �resultZaxis�outr� r r r �_median_nancheck� s
r� )N)r )NTFN)/r rI r= r| �rer- Znumpy.core.numerictypesr r r Znumpy.core.overridesr r r r r r r� Z numpy.compatr r
r� r
r# r$ r: r r r r rw r� r� r� r� r� r r r� �compile�Ir� r r� r� r r� r r r r �<module> sR
L49
i
3 +3 z
0 |