Working properly

Author: Thib99

numpy_questions.py

@@ -40,7 +40,14 @@ def max_index(X):
     i = 0
     j = 0
 
-    # TODO
+    if not isinstance(X, np.ndarray)  :
+        raise ValueError
+    if len(X.shape) != 2 :
+        raise ValueError
+        
+    index_linked = np.argmax(X).ravel()
+    
+    i, j = np.unravel_index(index_linked, X.shape)
 
     return i, j
 
@@ -64,4 +71,10 @@ def wallis_product(n_terms):
     """
     # XXX : The n_terms is an int that corresponds to the number of
     # terms in the product. For example 10000.
-    return 0.
+    if (n_terms==0) :
+        pi = 1
+    else :
+        n = np.arange(1, n_terms +1, dtype=np.float64)  # Generate n = 1, 2, ..., n_terms
+        ratios = (4 * n**2) / ((2 * n - 1) * (2 * n + 1))
+        pi = np.prod(ratios)
+    return pi*2