added numpy version

Author: PythonCHB

source/examples/threading-multiprocessing/integrate/integrate.py

@@ -10,6 +10,24 @@ def integrate(f, a, b, N):
     return s * dx
 
 
-def integrate_f_with_functional_tools(a, b, N):
-    dx = (float(b) - a) / N
+def integrate_f_with_functional_tools(f, a, b, N):
+    dx = (b - a) / N
     return sum(map(f, ((a + y * dx) for y in range(N)))) * dx
+
+
+def integrate_numpy(f, a, b, N):
+    """
+    numpy can be used to "vectorize" the problem
+
+    f must be "numpy comaptible"
+
+    """
+    # imported here so the rest of the code can run without it
+    import numpy as np
+    dx = (b - a) / N
+    i = np.arange(N)
+    s = np.sum(f(a + (i * dx)))
+    return s * dx
+
+
+

source/examples/threading-multiprocessing/integrate_main_profiler.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+
+from integrate.integrate import (f,
+                                 integrate_numpy as _integrate_numpy,
+                                 integrate as _integrate
+                                 )
+from decorators import timer
+
+
+@timer
+def integrate(*args):
+    return _integrate(*args)
+
+
+a = 0.0
+b = 10.0
+
+print("running the pure python version")
+for N in (10**i for i in range(1, 8)):
+    print("Numerical solution with N=%(N)d : %(x)f" %
+          {'N': N, 'x': integrate(f, a, b, N)})
+
+
+# now the numpy version:
+@timer
+def integrate_numpy(*args):
+    return _integrate_numpy(*args)
+
+
+print("running the numpy version")
+for N in (10**i for i in range(1, 8)):
+    print("Numerical solution with N=%(N)d : %(x)f" %
+          {'N': N, 'x': integrate_numpy(f, a, b, N)})

source/examples/threading-multiprocessing/integrate_profiler.py

@@ -6,6 +6,7 @@
 # sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
 
 from integrate.integrate import integrate, f
+# from integrate.integrate import f, integrate_numpy as integrate
 from decorators import timer