summaryrefslogtreecommitdiff
path: root/numpy/core/arrayprint.py
diff options
context:
space:
mode:
authorEric Wieser <wieser.eric@gmail.com>2017-11-04 23:45:01 -0700
committerEric Wieser <wieser.eric@gmail.com>2017-11-04 23:45:01 -0700
commit1b675d6c692a28fb12d597ff475d56f9a865e8b7 (patch)
tree53596674d02b2533ef1823a185b4730abfc35909 /numpy/core/arrayprint.py
parentd60709942188fc1b6edc738a0409b5e3ea0343d5 (diff)
downloadnumpy-1b675d6c692a28fb12d597ff475d56f9a865e8b7.tar.gz
MAINT: Rename formatters to match scalar type names
Diffstat (limited to 'numpy/core/arrayprint.py')
-rw-r--r--numpy/core/arrayprint.py38
1 files changed, 22 insertions, 16 deletions
diff --git a/numpy/core/arrayprint.py b/numpy/core/arrayprint.py
index 8253713be..2706d16f0 100644
--- a/numpy/core/arrayprint.py
+++ b/numpy/core/arrayprint.py
@@ -81,7 +81,7 @@ def _make_options_dict(precision=None, threshold=None, edgeitems=None,
if sign not in [None, '-', '+', ' ', 'legacy']:
raise ValueError("sign option must be one of "
"' ', '+', '-', or 'legacy'")
-
+
modes = ['fixed', 'unique', 'maxprec', 'maxprec_equal']
if floatmode not in modes + [None]:
raise ValueError("floatmode option must be one of " +
@@ -288,17 +288,18 @@ def _get_formatdict(data, **opt):
supp, sign = opt['suppress'], opt['sign']
# wrapped in lambdas to avoid taking a code path with the wrong type of data
- formatdict = {'bool': lambda: BoolFormat(data),
- 'int': lambda: IntegerFormat(data),
- 'float': lambda: FloatingFormat(data, prec, fmode,
- supp, sign),
- 'complexfloat': lambda: ComplexFormat(data, prec, fmode,
- supp, sign),
- 'datetime': lambda: DatetimeFormat(data),
- 'timedelta': lambda: TimedeltaFormat(data),
- 'object': lambda: _object_format,
- 'numpystr': lambda: repr_format,
- 'str': lambda: str}
+ formatdict = {
+ 'bool': lambda: BoolFormat(data),
+ 'int': lambda: IntegerFormat(data),
+ 'float': lambda:
+ FloatingFormat(data, prec, fmode, supp, sign),
+ 'complexfloat': lambda:
+ ComplexFloatingFormat(data, prec, fmode, supp, sign),
+ 'datetime': lambda: DatetimeFormat(data),
+ 'timedelta': lambda: TimedeltaFormat(data),
+ 'object': lambda: _object_format,
+ 'numpystr': lambda: repr_format,
+ 'str': lambda: str}
# we need to wrap values in `formatter` in a lambda, so that the interface
# is the same as the above values.
@@ -634,7 +635,10 @@ def _formatArray(a, format_function, rank, max_line_len,
summary_insert).rstrip()+']\n'
return s
+
class FloatingFormat(object):
+ """ Formatter for subtypes of np.floating """
+
def __init__(self, data, precision, floatmode, suppress_small, sign=False):
# for backcompatibility, accept bools
if isinstance(sign, bool):
@@ -644,7 +648,7 @@ class FloatingFormat(object):
if sign == 'legacy':
self._legacy = True
sign = '-' if data.shape == () else ' '
-
+
self.floatmode = floatmode
if floatmode == 'unique':
self.precision = -1
@@ -937,15 +941,17 @@ class BoolFormat(object):
return self.truestr if x else "False"
-class ComplexFormat(object):
+class ComplexFloatingFormat(object):
+ """ Formatter for subtypes of np.complexfloating """
+
def __init__(self, x, precision, floatmode, suppress_small, sign=False):
# for backcompatibility, accept bools
if isinstance(sign, bool):
sign = '+' if sign else '-'
- self.real_format = FloatingFormat(x.real, precision, floatmode,
+ self.real_format = FloatingFormat(x.real, precision, floatmode,
suppress_small, sign=sign)
- self.imag_format = FloatingFormat(x.imag, precision, floatmode,
+ self.imag_format = FloatingFormat(x.imag, precision, floatmode,
suppress_small, sign='+')
def __call__(self, x):