Xc @`sdZddlmZmZmZdddgZdZddlZddlm Z d d l m Z d d l mZmZmZmZmZmZd d lmZmZmZmZmZd d lmZd dlmZejddkrejZ ej d Z!nej"Z ej" d Z!dZ#da$da%da&e'a(da)da*da+da-dddddddddZ.dZ/dZ0dZ1dZ2dZ3dZ4ddddZ5ddddde6dd Z7d!Z8d"Z9d#e:fd$YZ;d%Z<d&e:fd'YZ=d(e:fd)YZ>d*e:fd+YZ?d,e:fd-YZ@d.e:fd/YZAd0e:fd1YZBd2e:fd3YZCd4e:fd5YZDdS(6sXArray printing function $Id: arrayprint.py,v 1.9 2005/09/13 13:58:44 teoliphant Exp $ i(tdivisiontabsolute_importtprint_functiont array2stringtset_printoptionstget_printoptionstrestructuredtextN(treducei(t numerictypes(tmaximumtminimumtabsolutet not_equaltisnantisinf(tarraytformat_longfloattdatetime_as_stringt datetime_datatdtype(travel(tasarrayicC`s||S(N((txty((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pytproduct"siiiKtnantinfcC`s|dk r|an|dk r*|an|dk r?|an|dk rT|an|dk rk| an|dk r|an|dk r|an|adS(sv 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 - 'numpy_str' : 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' 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( tNonet _line_widtht_summaryThresholdt_summaryEdgeItemst_float_output_precisiont_float_output_suppress_smallt_nan_strt_inf_strt _formatter(t precisiont thresholdt edgeitemst linewidthtsuppresstnanstrtinfstrt formatter((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyR0sm              cC`s=tdtdtdtdtdtdtdtdt}|S( s  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(R)R*R+( tdictRRRRR R!R"R#(td((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs  cC`sAddlm}|jdkr_t|dtkrV|j|t |t f}q=|}nt|dtkrgttt|tD]}t||^q}|j gttt|tddD]}t|| ^qn2gtdt|D]}t||^q }|jt |}|S(Ni(tnumericiii( tR.tndimtlenRt concatenatetrangetmint_leading_trailingtextendttuple(tat_nctbtitl((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyR5s  2 =2cC`s|r dSdSdS(Ns TruetFalse((R((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt_boolFormatterscC`s t|S(N(trepr(R((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt repr_formatscC`si td6t|d6t|||d6t|d6t|||d6t|d6t|d6t|d6td 6t d 6}|dk rg|j D]}||dk r|^q}d |krx%|j D]}|d ||t IntegerFormatt FloatFormattLongFloatFormatt ComplexFormattLongComplexFormattDatetimeFormattTimedeltaFormatR@RJRtkeys(tdataR$tsuppress_smallR+t formatdicttktfkeystkey((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt_get_formatdicts@          /      c C`s|j}|jd k rg}xe|jD]Z}||}tt||||}||jd krut|}n|j|q(Wt |S|j } t ||||} t | t jr| dSt | t jrt | t jr| dS| dSnt | t jr6t | t jr+| dS| dSn|t | t jrmt | t jrb| dS| dSnEt | t jt jfr| dSt | t jr| d S| dSd S( s; find the right formatting function for the dtype_ RARHRBRDRCRFRERIRGN((RtfieldsRtnamest_get_format_functionRtshapetSubArrayFormattappendtStructureFormatttypeR^t issubclasst_nttbool_tintegert timedelta64tfloatingRDtcomplexfloatingt clongfloattunicode_tstring_t datetime64( RXR$RYR+tdtype_tformat_functionst field_namet field_valuestformat_functiontdtypeobjRZ((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRas@       t R/c C`s|jtkr$d}t|}nd}tt|}t||||} d} | dt|7} t|| t|j|| |t |d } | S(Ns..., R/Rxi( tsizeRR5RRRaR1t _formatArrayRbR( R8tmax_line_widthR$RYt separatortprefixR+tsummary_insertRXRvtnext_line_prefixtlst((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt _array2string;s  c C`s|dkrt}n|dkr*t}n|dkr?t}n|dkrTt}n|jdkr|j}|jjdk rt |gd|j} t | |||} | | d} q||} nBt t |jdkrd} n!t ||||||d|} | S(s 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 - 'numpy_str' : 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]' Ris[]R+N((RRRR R#RbtitemRR_RRaRRR( R8R{R$RYR|R}tstyleR+RtarrRvR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRTs*a          cC`sYt|jt|j|krE||jd7}|}n||7}||fS(Ns (R1trstrip(tstlinetwordt max_line_lenR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt _extendLines (  c C`s?|dkrtdn|rLd|t|krL|}|} |} nd}t|} d} |dkrd} |} xFt|D]8} ||| |}t| | |||\} } qW| rt| | | ||\} } nxMt| ddD]9} ||| |}t| | |||\} } qW||d}t| | |||\} } | | d7} d| t|} nd} |j}xt|D]{} | dkr| |7} n| t|| ||d|d ||||7} | j|jd t|dd} qW| rH| || d 7} nxt| ddD]} |ss| | kr| |7} n| t|| ||d|d ||||7} | j|jd t|dd} q[W|s| dkr| |7} n| t|d||d|d ||||jd7} | S( sgformatArray is designed for two modes of operation: 1. Full output 2. Summarized output isrank shouldn't be zero.iR/iis] t[Rxs (t ValueErrorR1R3RRRztmax(R8RvtrankRRR|t edge_itemsR~t leading_itemsttrailing_itemstsummary_insert1RRR;Rtsep((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRzs`    "!"     /   /  RQcB`s)eZedZdZedZRS(cC`se||_||_||_t|_t|_d|_y|j|Wntt fk r`nXdS(Ni( R$RYtsignR=t exp_formattlarge_exponentt max_str_lent fillFormatt TypeErrortNotImplementedError(tselfRXR$RYR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt__init__s      c C`sddlm}|jddt|t|B}t|d|@}t|j|}t|dkrd}d}nht j |}t j |}|dkrt |_ n|j r|dks||d krt |_ nWdQX|j rd|kod knp|d k|_d |j|_|jrP|jd7_n|jrbd }nd}|d|j|jf}nd|jf}t|rtg|D]} t| |j|^q} nd} t|j| } ttt|| d|_|j|rGt|jttttd|_n|jrYd}nd}|d|j| f}d|jf|_||_dS(Ni(R.RKtignoreiggחAg-C6?g@@g>N}a+g}Ô%ITis%+t%s%d.%des%%.%dfis%#+s%#s%d.%dfs%%%ds(R/R.terrstateR RR R tcompressR1R RR tTrueRRYRR$RRRt_digitsR4RJRBtanyR!R"t special_fmttformat( RRXR9tspecialtvalidtnon_zerotmax_valtmin_valRRR$((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyR-sR    +    +#    cC`sddlm}|jddt|r[|jrJ|jdtfS|jtfSnYt|r|dkr|jr|jdtfS|jtfSq|jdtfSnWdQX|j |}|j r|d}|dks|dkr||dd !d |d }q|nk|j rF|dd kr|d |d |d }q|n6|r||j d }|d t |t |}n|S( Ni(R.tinvalidRt+it-iit0Rx(R/R.RR RRR!RR"RRRRR1(RRt strip_zerosR9Rtexpsigntz((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyt__call__`s0         !(t__name__t __module__R=RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRQs  3cC`sE|dkr=||}|jd}|t|t|SdSdS(NiR(RR1(RR$RRR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs   RPcB`seZdZdZRS(cC`syTttttj|tttj|}dt|d|_Wn'ttfk rmnt k r}nXdS(NRR-( RR1RJR RR RRRR(RRXR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs cC`s3t|kotknr'|j|Sd|SdS(Ns%s(t_MININTt_MAXINTR(RR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs (RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRPs RRcB`seZedZdZRS(cC`s||_||_dS(N(R$R(RR$R((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs cC`st|r(|jrdtSdtSnt|rg|dkr\|jrQdtSdtSqdtSnP|dkr|jrdt||jSdt||jSnt||jSdS(NRRxiR(R RR!RR"RR$(RR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs          (RRR=RR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRRs RTcB`seZdZdZRS(cC`s(t||_t|dt|_dS(NR(RRt real_formatRt imag_format(RR$((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRscC`s0|j|j}|j|j}||dS(Ntj(RtrealRtimag(RRtrR;((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs(RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRTs RScB`seZdZdZRS(cC`s:t|j|||_t|j||dt|_dS(NR(RQRRRRR(RRR$RY((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRscC`s|j|jdt}|j|jdt}|jjsp|jd}|ddt|t|}n |d}||S(NRRRRx(RRR=RRRRR1(RRRR;R((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs % (RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRSs RUcB`s#eZddddZdZRS(t same_kindcC`sq|dkr=|jjdkr4t|jd}q=d}n|dkrRd}n||_||_||_dS(NtMiRtnaive(RRtkindRttimezonetunittcasting(RRRRR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs      c C`s)dt|d|jd|jd|jS(Ns'%s'RRR(RRRR(RR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs   N(RRRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRUsRVcB`seZdZdZRS(cC`s|jjdkrtdgd|jd}t|jjd}|j|}|t||j|}t|dkrtttt j |ttt j |}nd}t|t|krt|d}ndt|d|_ d j ||_ndS( NtmtNaTRiti8iRR-s'NaT'(RRRt byteordertviewR R1RRJR RR Rtrjustt_nat(RRXt nat_valuet int_dtypetint_viewtvR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs!cC`sA|djd|jdkr)|jS|j|jdSdS(NiR(RRRtastype(RR((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs"(RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRVs RccB`seZdZdZRS(cC`s ||_dS(N(Rv(RRv((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyR sc`sY|jdkr4ddjfd|DdSddjfd|DdS(NiRs, c3`s|]}j|VqdS(N(Rv(t.0R8(R(s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pys st]c3`s|]}j|VqdS(N(R(RR8(R(s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pys s(R0tjoin(RR((Rs4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyR s%(RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRc s RecB`seZdZdZRS(cC`s||_t||_dS(N(RsR1t num_fields(RRs((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs cC`sbd}x4t||jD] \}}|||d7}qWd|jkrV|d n|d dS(Nt(s, iiit)(tzipRsR(RRRtfieldRv((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRs(RRRR(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pyRes (Et__doc__t __future__RRRt__all__t __docformat__tsyst functoolsRR/RRhtumathR R R R R Rt multiarrayRRRRRt fromnumericRR.Rt version_infotmaxsizeRRtmaxintRRRRR=R RR!R"RR#RRR5R>R@R^RaRR?RRRztobjectRQRRPRRRTRSRURVRcRe(((s4/tmp/pip-build-X4mzal/numpy/numpy/core/arrayprint.pytsf .(     z #    $ * }  Cd