MAINT: Remove Python2 workarounds

1 files changed, 2 insertions, 8 deletions

diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py
index d2f26149b..e909cec6f 100644
--- a/numpy/core/shape_base.py
+++ b/numpy/core/shape_base.py
@@ -2,6 +2,7 @@ __all__ = ['atleast_1d', 'atleast_2d', 'atleast_3d', 'block', 'hstack',
            'stack', 'vstack']
 
 import functools
+import itertools
 import operator
 import warnings
 
@@ -530,14 +531,7 @@ def _atleast_nd(a, ndim):
 
 
 def _accumulate(values):
-    # Helper function because Python 2.7 doesn't have
-    # itertools.accumulate
-    value = 0
-    accumulated = []
-    for v in values:
-        value += v
-        accumulated.append(value)
-    return accumulated
+    return list(itertools.accumulate(values))
 
 
 def _concatenate_shapes(shapes, axis):