diff options
Diffstat (limited to 'numpy/core/arrayprint.py')
| -rw-r--r-- | numpy/core/arrayprint.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/numpy/core/arrayprint.py b/numpy/core/arrayprint.py index 93a659616..7b2b37694 100644 --- a/numpy/core/arrayprint.py +++ b/numpy/core/arrayprint.py @@ -273,25 +273,24 @@ def get_printoptions(): return _format_options.copy() -def _leading_trailing(a, index=()): +def _leading_trailing(a, edgeitems, index=()): """ Keep only the N-D corners (leading and trailing edges) of an array. Should be passed a base-class ndarray, since it makes no guarantees about preserving subclasses. """ - edgeitems = _format_options['edgeitems'] axis = len(index) if axis == a.ndim: return a[index] if a.shape[axis] > 2*edgeitems: return concatenate(( - _leading_trailing(a, index + np.index_exp[ :edgeitems]), - _leading_trailing(a, index + np.index_exp[-edgeitems:]) + _leading_trailing(a, edgeitems, index + np.index_exp[ :edgeitems]), + _leading_trailing(a, edgeitems, index + np.index_exp[-edgeitems:]) ), axis=axis) else: - return _leading_trailing(a, index + np.index_exp[:]) + return _leading_trailing(a, edgeitems, index + np.index_exp[:]) def _object_format(o): @@ -437,7 +436,7 @@ def _array2string(a, options, separator=' ', prefix=""): if a.size > options['threshold']: summary_insert = "..." - data = _leading_trailing(data) + data = _leading_trailing(data, options['edgeitems']) else: summary_insert = "" |