2 files changed, 8 insertions, 0 deletions
diff --git a/numpy/lib/arrayterator.py b/numpy/lib/arrayterator.py
index d4a91b001..3f07cd263 100644
--- a/numpy/lib/arrayterator.py
+++ b/numpy/lib/arrayterator.py
@@ -14,6 +14,10 @@ from operator import mul
 
 __all__ = ['Arrayterator']
 
+import sys
+if sys.version_info[0] >= 3:
+    from functools import reduce
+
 class Arrayterator(object):
     """
     Buffered iterator for big arrays.
diff --git a/numpy/lib/tests/test_arrayterator.py b/numpy/lib/tests/test_arrayterator.py
index c600d45eb..3dce009d3 100644
--- a/numpy/lib/tests/test_arrayterator.py
+++ b/numpy/lib/tests/test_arrayterator.py
@@ -4,6 +4,10 @@ import numpy as np
 from numpy.random import randint
 from numpy.lib import Arrayterator
 
+import sys
+if sys.version_info[0] >= 3:
+    from functools import reduce
+
 def test():
     np.random.seed(np.arange(10))