Simplify code

Author: gjbex

source-code/cython/Numpy/array_init.pyx

@@ -1,9 +1,8 @@
-from functools import reduce
 import numpy as np
 
 
 def cons(dims):
-    array = np.empty(reduce(lambda x, y: x*y, dims, 1))
+    array = np.empty(np.prod(dims))
     _cons(memoryview(array))
     return array.reshape(dims)
 

source-code/cython/Numpy/array_sum.pyx

@@ -2,16 +2,14 @@ from cython cimport boundscheck, wraparound
 
 
 def array_sum(a):
-    return _array_sum(memoryview(a))
+    return _array_sum(memoryview(a.reshape((-1, ))))
 
 
-cdef double _array_sum(double[:,::1] mem_view):
+cdef double _array_sum(double[:] mem_view):
     cdef double total = 0.0
     cdef m = mem_view.shape[0]
-    cdef n = mem_view.shape[1]
-    cdef int i, j
+    cdef int i
     with boundscheck(False), wraparound(False):
         for i in range(m):
-            for j in range(n):
-                total += mem_view[i, j]
+            total += mem_view[i]
     return total

source-code/cython/Numpy/compute_sums.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python
 
 from argparse import ArgumentParser
+import array_sum
 import numpy as np
 import timeit
-from array_sum import array_sum
 
 
 def py_sum(a):
@@ -16,17 +16,16 @@ def py_sum(a):
 if __name__ == '__main__':
     arg_parser = ArgumentParser(description='compute array sum using '
                                             'np.sum, cython and python')
-    arg_parser.add_argument('--n', type=int, default=10000,
+    arg_parser.add_argument('--n', type=int, default=10_000,
                             help='size (nxn array)')
     arg_parser.add_argument('--iter', type=int, default=10,
                             help='number of iterations')
     options = arg_parser.parse_args()
     a = np.ones((options.n, options.n))
-    for func in [array_sum, np.sum, py_sum]:
+    for func in [array_sum.array_sum, np.sum, py_sum]:
         total = 0.0
         start_time = timeit.default_timer()
         for iter_nr in range(options.iter):
             total += func(a)
         total_time = timeit.default_timer() - start_time
-        print('{0:s}: {1:.6f} s ({2})'.format(func.__name__, total_time,
-                                              total))
+        print(f'{func.__name__:s}: {total_time:.6f} s ({total})')