From ffa569766c48bf9ec2df0fb7b407707511ec878d Mon Sep 17 00:00:00 2001 From: Rayyan Ahmed Date: Fri, 20 Dec 2024 21:35:28 +0530 Subject: [PATCH] UP my solution --- numpy_questions.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/numpy_questions.py b/numpy_questions.py index 07a10c1..1ec90e0 100644 --- a/numpy_questions.py +++ b/numpy_questions.py @@ -39,9 +39,15 @@ def max_index(X): """ i = 0 j = 0 + if not isinstance(X, np.ndarray): + raise ValueError("Input must be a numpy array.") - # TODO + if X.ndim != 2: + raise ValueError("Input array must be 2D.") + # Find the index of the maximum value + max_idx = np.unravel_index(np.argmax(X), X.shape) + i, j = max_idx return i, j @@ -64,4 +70,12 @@ 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: + raise ValueError("Number of terms must be non-negative.") + + product = 1.0 + for n in range(1, n_terms + 1): + term = (4 * n**2) / ((4 * n**2) - 1) + product *= term + + return 2 * product