Update numpy_questions.py

Author: Ahmed-Semah

numpy_questions.py

@@ -20,7 +20,6 @@
 
 def max_index(X):
     """Return the index of the maximum in a numpy array.
-    dsvsdf
     Parameters
     ----------
     X : ndarray of shape (n_samples, n_features)
@@ -41,8 +40,13 @@ def max_index(X):
     j = 0
 
     # TODO
-
+    if not isinstance(X, np.ndarray):
+        raise ValueError("Input is not a numpy array")
+    if X.ndim != 2:
+        raise ValueError("Input is not a 2D array")
+    i, j = np.unravel_index(np.argmax(X), X.shape)
     return i, j
+    
 
 
 def wallis_product(n_terms):