Edit File: arrayprint.cpython-37.pyc
B ��Fd r � @ s� d Z ddlmZmZmZ dddgZdZddlZddlZej d dkrxydd l m Z W q� e k rt dd l m Z Y q�X n2ydd lm Z W n e k r� dd lm Z Y nX d d lmZ d d lmZmZmZmZmZmZ d d lmZm Z m Z m Z m Z d dl m!Z! d dl"m#Z# ej d dk�r.ej$Z%ej$ d Z&nej'Z%ej' d Z&dd� Z(da)da*da+da,da-da.da/da0dFdd�Z1dd� Z2dd� Z3d d � Z4d d � Z5d d!� Z6d"d#� Z7d$d%� Z8dGd(d)�Z9dHd+d,�Z:e:� dddd&d'e;dfd-d��Z<d.d/� Z=d0d1� Z>G d2d3� d3e?�Z@d4d5� ZAG d6d7� d7e?�ZBG d8d9� d9e?�ZCG d:d;� d;e?�ZDG d<d=� d=e?�ZEG d>d?� d?e?�ZFG d@dA� dAe?�ZGG dBdC� dCe?�ZHG dDdE� dEe?�ZIdS )IzXArray printing function $Id: arrayprint.py,v 1.9 2005/09/13 13:58:44 teoliphant Exp $ � )�division�absolute_import�print_function� array2string�set_printoptions�get_printoptions�restructuredtextN� )� get_ident� )� numerictypes)�maximum�minimum�absolute� not_equal�isnan�isinf)�array�format_longfloat�datetime_as_string� datetime_data�dtype)�ravel)�asarrayc C s | | S )N� )�x�yr r �H/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/arrayprint.py�product- s r i� � F�K �nan�infc C s` |dk r |a |dk r|a|dk r$|a| dk r0| a|dk r@| a|dk rL|a|dk rX|a|adS )a� Set printing options. These options determine the way floating point numbers, arrays and other NumPy objects are displayed. Parameters ---------- precision : int, optional Number of digits of precision for floating point output (default 8). threshold : int, optional Total number of array elements which trigger summarization rather than full repr (default 1000). edgeitems : int, optional Number of array items in summary at beginning and end of each dimension (default 3). linewidth : int, optional The number of characters per line for the purpose of inserting line breaks (default 75). suppress : bool, optional Whether or not suppress printing of small floating point values using scientific notation (default False). nanstr : str, optional String representation of floating point not-a-number (default nan). infstr : str, optional String representation of floating point infinity (default inf). formatter : dict of callables, optional If not None, the keys should indicate the type(s) that the respective formatting function applies to. Callables should return a string. Types that are not specified (by their corresponding keys) are handled by the default formatters. Individual types for which a formatter can be set are:: - 'bool' - 'int' - 'timedelta' : a `numpy.timedelta64` - 'datetime' : a `numpy.datetime64` - 'float' - 'longfloat' : 128-bit floats - 'complexfloat' - 'longcomplexfloat' : composed of two 128-bit floats - 'numpystr' : types `numpy.string_` and `numpy.unicode_` - 'object' : `np.object_` arrays - 'str' : all other strings Other keys that can be used to set a group of types at once are:: - 'all' : sets all types - 'int_kind' : sets 'int' - 'float_kind' : sets 'float' and 'longfloat' - 'complex_kind' : sets 'complexfloat' and 'longcomplexfloat' - 'str_kind' : sets 'str' and 'numpystr' See Also -------- get_printoptions, set_string_function, array2string Notes ----- `formatter` is always reset with a call to `set_printoptions`. Examples -------- Floating point precision can be set: >>> np.set_printoptions(precision=4) >>> print(np.array([1.123456789])) [ 1.1235] Long arrays can be summarised: >>> np.set_printoptions(threshold=5) >>> print(np.arange(10)) [0 1 2 ..., 7 8 9] Small results can be suppressed: >>> eps = np.finfo(float).eps >>> x = np.arange(4.) >>> x**2 - (x + eps)**2 array([ -4.9304e-32, -4.4409e-16, 0.0000e+00, 0.0000e+00]) >>> np.set_printoptions(suppress=True) >>> x**2 - (x + eps)**2 array([-0., -0., 0., 0.]) A custom formatter can be used to display array elements as desired: >>> np.set_printoptions(formatter={'all':lambda x: 'int: '+str(-x)}) >>> x = np.arange(3) >>> x array([int: 0, int: -1, int: -2]) >>> np.set_printoptions() # formatter gets reset >>> x array([0, 1, 2]) To put back the default options, you can use: >>> np.set_printoptions(edgeitems=3,infstr='inf', ... linewidth=75, nanstr='nan', precision=8, ... suppress=False, threshold=1000, formatter=None) N)� _line_width�_summaryThreshold�_summaryEdgeItems�_float_output_precision� _float_output_suppress_small�_nan_str�_inf_str� _formatter)� precision� threshold� edgeitems� linewidth�suppress�nanstr�infstr� formatterr r r r ; s nc C s t ttttttttd�} | S )a Return the current print options. Returns ------- print_opts : dict Dictionary of current print options with keys - precision : int - threshold : int - edgeitems : int - linewidth : int - suppress : bool - nanstr : str - infstr : str - formatter : dict of callables For a full description of these options, see `set_printoptions`. See Also -------- set_printoptions, set_string_function )r+ r, r- r. r/ r0 r1 r2 ) �dictr& r$ r% r# r' r( r) r* )�dr r r r � s c s� ddl m} � jdkrNt� �dt krH|�� d t� � t d � f�}qԈ }n�t� �dt kr�� fdd�ttt� �t��D �}|�� fdd�ttt� �t�dd�D �� n � fd d�tdt� ��D �}|�t |��}|S ) Nr )�numeric� c s g | ]}t � | ��qS r )�_leading_trailing)�.0�i)�ar r � <listcomp>� s z%_leading_trailing.<locals>.<listcomp>c s g | ]}t � | ��qS r )r7 )r8 r9 )r: r r r; � s r ���c s g | ]}t � | ��qS r )r7 )r8 r9 )r: r r r; � s ) � r5 �ndim�lenr% Z concatenate�range�min�extend�tuple)r: �_nc�b�lr )r: r r7 � s r7 c C s | rdS dS d S )Nz True�Falser )r r r r �_boolFormatter� s rH c C s t | �tkrd}nd}|�| �S )z@ Object arrays containing lists should be printed unambiguously z list({!r})z{!r})�type�list�format)�o�fmtr r r �_object_format� s rN c C s t | �S )N)�repr)r r r r � repr_format� s rP c s� dd� � fdd�� ��fdd��fdd�� ��fdd��fdd�� fdd�� fd d�d d� d d� d d� d � }dd� }�d k �r��fdd��� � D �}d|kr�x |� � D ]}|�d �||< q�W d|kr�x dD ]}|�d �||< q�W d|k�r x dD ]}|�d �||< q�W d|k�r6x dD ]}|�d �||< �q W d|k�r`x dD ]}|�d �||< �qFW x,|� � D ] }||k�rj|�| �||< �qjW |S )Nc S s t S )N)rH r r r r �<lambda> � z!_get_formatdict.<locals>.<lambda>c s t � �S )N)� IntegerFormatr )�datar r rQ rR c s t � ���S )N)� FloatFormatr )rT r+ �suppress_smallr r rQ rR c s t � �S )N)�LongFloatFormatr )r+ r r rQ rR c s t � ���S )N)� ComplexFormatr )rT r+ rV r r rQ s c s t � �S )N)�LongComplexFormatr )r+ r r rQ rR c s t � �S )N)�DatetimeFormatr )rT r r rQ rR c s t � �S )N)�TimedeltaFormatr )rT r r rQ rR c S s t S )N)rN r r r r rQ rR c S s t S )N)rP r r r r rQ rR c S s t S )N)�strr r r r rQ rR ) �bool�int�float� longfloat� complexfloat�longcomplexfloat�datetime� timedelta�object�numpystrr\ c s � fdd�S )Nc s � S )Nr r )r r r rQ rR z3_get_formatdict.<locals>.indirect.<locals>.<lambda>r )r r )r r �indirect s z!_get_formatdict.<locals>.indirectc s g | ]}� | d k r|�qS )Nr )r8 �k)r2 r r r; s z#_get_formatdict.<locals>.<listcomp>�allZint_kind)r^ Z float_kind)r_ r` Z complex_kind)ra rb Zstr_kind)rf r\ )�keys)rT r+ rV r2 � formatdictrg Zfkeys�keyr )rT r2 r+ rV r �_get_formatdict sB rm c C sz | j }|jdk rfg }xH|jD ]>}| | }tt|�|||�}|| jdkrPt|�}|�|� q W t|�S |j } t | |||�} t | t j �r�| d � S t | t j�r�t | t j�r�| d � S | d � S n�t | t j�r�t | t j�r�| d � S | d � S n�t | t j��r t | t j��r| d � S | d � S nXt | t jt jf��r<| d � S t | t j��rT| d � S t | t j��rl| d � S | d � S dS ) z; find the right formatting function for the dtype_ Nr r] rd r^ r` r_ rb ra rf rc re )r �fields�names�_get_format_functionr �shape�SubArrayFormat�append�StructureFormatrI rm � issubclass�_ntZbool_�integerZ timedelta64Zfloatingr` ZcomplexfloatingZ clongfloatZunicode_Zstring_Z datetime64Zobject_) rT r+ rV r2 Zdtype_�format_functions� field_nameZ field_values�format_functionZdtypeobjrk r r r rp + sD rp � r= c C sn | j tkrd}t| �}nd}tt| ��}t||||�} d} | dt|� 7 } t| | | j|| |t |�d d� } | S )Nz..., r= r{ r< ) �sizer$ r7 r r rp r? � _formatArrayr> r% ) r: �max_line_widthr+ rV � separator�prefixr2 �summary_insertrT rz �next_line_prefix�lstr r r � _array2stringW s r� �...c s � fdd�}|S )a Like the python 3.2 reprlib.recursive_repr, but forwards *args and **kwargs Decorates a function such that if it calls itself with the same first argument, it returns `fillvalue` instead of recursing. Largely copied from reprlib.recursive_repr c s$ t � �t�� �� ��fdd��}|S )Nc sF t | �t� f}|�kr�S ��|� z� | f|�|�S ��|� X d S )N)�idr �add�discard)�self�args�kwargsrl )�f� fillvalue� repr_runningr r �wrapper} s z>_recursive_guard.<locals>.decorating_function.<locals>.wrapper)�set� functools�wraps)r� r� )r� )r� r� r �decorating_functionz s z-_recursive_guard.<locals>.decorating_functionr )r� r� r )r� r �_recursive_guardp s r� c C s� |dkr t }|dkrt}|dkr$t}|dkr0t}| jdkr�| �� }| jjdk rzt|g| jd�} t | |||�} | | d �} q�||�} n.t � t | j�dkr�d} nt | ||||||d�} | S )a� Return a string representation of an array. Parameters ---------- a : ndarray Input array. max_line_width : int, optional The maximum number of columns the string should span. Newline characters splits the string appropriately after array elements. precision : int, optional Floating point precision. Default is the current printing precision (usually 8), which can be altered using `set_printoptions`. suppress_small : bool, optional Represent very small numbers as zero. A number is "very small" if it is smaller than the current printing precision. separator : str, optional Inserted between elements. prefix : str, optional An array is typically printed as:: 'prefix(' + array2string(a) + ')' The length of the prefix string is used to align the output correctly. style : function, optional A function that accepts an ndarray and returns a string. Used only when the shape of `a` is equal to ``()``, i.e. for 0-D arrays. formatter : dict of callables, optional If not None, the keys should indicate the type(s) that the respective formatting function applies to. Callables should return a string. Types that are not specified (by their corresponding keys) are handled by the default formatters. Individual types for which a formatter can be set are:: - 'bool' - 'int' - 'timedelta' : a `numpy.timedelta64` - 'datetime' : a `numpy.datetime64` - 'float' - 'longfloat' : 128-bit floats - 'complexfloat' - 'longcomplexfloat' : composed of two 128-bit floats - 'numpystr' : types `numpy.string_` and `numpy.unicode_` - 'str' : all other strings Other keys that can be used to set a group of types at once are:: - 'all' : sets all types - 'int_kind' : sets 'int' - 'float_kind' : sets 'float' and 'longfloat' - 'complex_kind' : sets 'complexfloat' and 'longcomplexfloat' - 'str_kind' : sets 'str' and 'numpystr' Returns ------- array_str : str String representation of the array. Raises ------ TypeError if a callable in `formatter` does not return a string. See Also -------- array_str, array_repr, set_printoptions, get_printoptions Notes ----- If a formatter is specified for a certain type, the `precision` keyword is ignored for that type. This is a very flexible function; `array_repr` and `array_str` are using `array2string` internally so keywords with the same name should work identically in all three functions. Examples -------- >>> x = np.array([1e-16,1,2,3]) >>> print(np.array2string(x, precision=2, separator=',', ... suppress_small=True)) [ 0., 1., 2., 3.] >>> x = np.arange(3.) >>> np.array2string(x, formatter={'float_kind':lambda x: "%.2f" % x}) '[0.00 1.00 2.00]' >>> x = np.arange(3) >>> np.array2string(x, formatter={'int':lambda x: hex(x)}) '[0x0L 0x1L 0x2L]' Nr )r r z[])r2 )r# r& r' r* rq �itemr rn r rp r� �reducer r� ) r: r~ r+ rV r r� �styler2 r �arrrz r� r r r r � s* b c C s@ t |�� �t |�� � |kr0| |�� d 7 } |}||7 }| |fS )N� )r? �rstrip)�s�line�word� max_line_lenr� r r r � _extendLine s r� c C sv |dkrt d��|r2d| t| �k r2|}|} |} nd}t| �} d} |dk�r"d} |} x4t|�D ](} || | �| }t| | |||�\} } q^W | r�t| | | ||�\} } x:t| dd�D ]*} || | �| }t| | |||�\} } q�W || d �}t| | |||�\} } | | d 7 } d| t|�d � } �nPd} |�� }xlt|�D ]`} | dk�rN| |7 } | t| | ||d |d | |||�7 } | �� |�� d t|d d� } �q8W | �r�| || d 7 } xxt| dd�D ]h} |�s�| | k�r�| |7 } | t| | ||d |d | |||�7 } | �� |�� d t|d d� } �q�W |�s<| dk�rD| |7 } | t| d ||d |d | |||��� d 7 } | S ) zgformatArray is designed for two modes of operation: 1. Full output 2. Summarized output r zrank shouldn't be zero.r6 r= r r< z] �[Nr{ r� )� ValueErrorr? r@ r� r� r} �max)r: rz Zrankr� r� r Z edge_itemsr� Z leading_itemsZtrailing_itemsZsummary_insert1r� r� r9 r� �sepr r r r} s` ( ( r} c @ s( e Zd Zd dd�Zdd� Zd dd�Zd S ) rU Fc C sP || _ || _|| _d| _d| _d| _y| �|� W n ttfk rJ Y nX d S )NFr ) r+ rV �sign� exp_format�large_exponent� max_str_len� fillFormat� TypeError�NotImplementedError)r� rT r+ rV r� r r r �__init__[ s zFloatFormat.__init__c s� ddl m} |jdd��� t|�t|�B }t|d�| @ }t|�|��}t|�dkr^d}d}nBt � |�}t � |�}|dkr�d�_ �j s�|d k s�|| d kr�d�_ W d Q R X �j �r d| k o�d k n p�|d k�_d �j �_�jr� jd7 _�j�rd� nd� � d�j�jf � n�d�jf � t|��rLt� �fdd�|D ��}nd}t�j|�}ttt|���| d �_|�|��r�t�jtt�tt�d ��_�j�r�d� nd� � d�j|f � d�jf �_� �_d S )Nr )r5 �ignore)ri r g g �חATg-C ��6?g @�@g>��N}a+g}Ô%�I�Tr z%+�%z%d.%dez%%.%dfc s g | ]}t |�j� ��qS r )�_digitsr+ )r8 r )rK r� r r r; � s z*FloatFormat.fillFormat.<locals>.<listcomp>r6 z%#+z%#z%d.%dfz%%%ds)r= r5 �errstater r r r �compressr? r r� r r� rV r� r+ r� r� r� rA r\ r^ �anyr( r) � special_fmtrK ) r� rT rD �specialZvalidZnon_zeroZmax_valZmin_valr+ r )rK r� r r� i sR " zFloatFormat.fillFormatTc C s: ddl m} |jdd��v t|�rF| jr8| jdt f S | jtf S nDt|�r�|dkrz| jrl| jdt f S | jtf S n| jdt f S W d Q R X | j | }| j r�|d }|dks�|dkr�|dd � d |d d � }n\| j �r|d d k�r6d |d d� |d d � }n(|�r6|� d �}|d t |�t |� }|S ) Nr )r5 r� )�invalid�+r �-�������0r{ )r= r5 r� r r� r� r( r r) rK r� r� r� r? )r� r � strip_zerosrD r� Zexpsign�zr r r �__call__� s0 zFloatFormat.__call__N)F)T)�__name__� __module__� __qualname__r� r� r� r r r r rU Z s 3rU c C s6 |dkr.|| }|� d�}|t|� t|� S dS d S )Nr r� )r� r? )r r+ rK r� r� r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )rS c C sl y<t ttt�|���ttt�|����}dt|� d | _W n* ttfk rT Y n t k rf Y nX d S )Nr� r4 ) r� r? r\ r r� r rK r� r� r� )r� rT r� r r r r� � s zIntegerFormat.__init__c C s. t | k rtk r"n n | j| S d| S d S )Nz%s)�_MININT�_MAXINTrK )r� r r r r r� � s zIntegerFormat.__call__N)r� r� r� r� r� r r r r rS � s rS c @ s e Zd Zddd�Zdd� ZdS )rW Fc C s || _ || _d S )N)r+ r� )r� r+ r� r r r r� � s zLongFloatFormat.__init__c C s� t |�r | jrdt S dt S nnt|�rR|dkrH| jr>dt S dt S q�dt S n<|dkr�| jrpdt|| j� S dt|| j� S n t|| j�S d S )Nr� r{ r r� )r r� r( r r) r r+ )r� r r r r r� � s zLongFloatFormat.__call__N)F)r� r� r� r� r� r r r r rW � s rW c @ s e Zd Zdd� Zdd� ZdS )rY c C s t |�| _t |dd�| _d S )NT)r� )rW � real_format� imag_format)r� r+ r r r r� � s zLongComplexFormat.__init__c C s$ | � |j�}| �|j�}|| d S )N�j)r� �realr� �imag)r� r �rr9 r r r r� � s zLongComplexFormat.__call__N)r� r� r� r� r� r r r r rY � s rY c @ s e Zd Zdd� Zdd� ZdS )rX c C s( t |j||�| _t |j||dd�| _d S )NT)r� )rU r� r� r� r� )r� r r+ rV r r r r� s zComplexFormat.__init__c C s` | j |jdd�}| j|jdd�}| jjsP|�d�}|d dt|�t|� }n|d }|| S )NF)r� r� r� r{ )r� r� r� r� r� r� r? )r� r r� r9 r� r r r r� s zComplexFormat.__call__N)r� r� r� r� r� r r r r rX s rX c @ s e Zd Zddd�Zdd� ZdS )rZ N� same_kindc C sJ |d kr(|j jdkr$t|j �d }nd}|d kr4d}|| _|| _|| _d S )N�Mr r� Znaive)r �kindr �timezone�unit�casting)r� r r� r� r� r r r r� s zDatetimeFormat.__init__c C s dt || j| j| jd� S )Nz'%s')r� r� r� )r r� r� r� )r� r r r r r� $ s zDatetimeFormat.__call__)NNr� )r� r� r� r� r� r r r r rZ s rZ c @ s e Zd Zdd� Zdd� ZdS )r[ c C s� |j jdkr�tdg|j d�d }t |j jd �}|�|�}|t||�|�� }t|�dkr�tttt � |���ttt � |����}nd}t|�t|�k r�t|d�}dt|� d | _ d � |�| _d S ) N�mZNaT)r r �i8� r� r4 z'NaT')r r� r � byteorder�viewr r? r� r\ r r� r rK �rjust�_nat)r� rT Z nat_valueZ int_dtypeZint_view�vr� r r r r� + s zTimedeltaFormat.__init__c C s2 |d � d�|� d�kr | jS | j|�d� S d S )Nr r� )r� r� rK �astype)r� r r r r r� = s zTimedeltaFormat.__call__N)r� r� r� r� r� r r r r r[ * s r[ c @ s e Zd Zdd� Zdd� ZdS )rr c C s || _ d S )N)rz )r� rz r r r r� F s zSubArrayFormat.__init__c sJ |j dkr*dd�� fdd�|D �� d S dd�� fdd�|D �� d S )Nr r� z, c 3 s | ]}� � |�V qd S )N)rz )r8 r: )r� r r � <genexpr>K s z*SubArrayFormat.__call__.<locals>.<genexpr>�]c 3 s | ]}� � |�V qd S )N)r� )r8 r: )r� r r r� L s )r> �join)r� r� r )r� r r� I s zSubArrayFormat.__call__N)r� r� r� r� r� r r r r rr E s rr c @ s e Zd Zdd� Zdd� ZdS )rt c C s || _ t|�| _d S )N)rx r? � num_fields)r� rx r r r r� P s zStructureFormat.__init__c C sT d}x(t || j�D ]\}}|||�d 7 }qW d| jk rD|d d� n |d d� d S )N�(z, r r� r< �))�ziprx r� )r� r r� �fieldrz r r r r� T s zStructureFormat.__call__N)r� r� r� r� r� r r r r rt O s rt )NNNNNNNN)r{ r= N)r� )J�__doc__� __future__r r r �__all__� __docformat__�sysr� � version_info�_threadr � ImportError� _dummy_thread�thread� dummy_threadr= r rv �umathr r r r r r � multiarrayr r r r r Z fromnumericr r5 r �maxsizer� r� Zmaxintr r% r$ r& r' r# r( r) r* r r r7 rH rN rP rm rp r� r� rO r r� r} re rU r� rS rW rY rX rZ r[ rr rt r r r r �<module> s~ {#+, }Cd