From 7c2448f2da5d8c8ab1efb2397107571d871f9a95 Mon Sep 17 00:00:00 2001 From: ASPP Student Date: Tue, 23 Sep 2025 16:23:16 +0300 Subject: [PATCH] Initial version of local_maxima --- hands_on/local_maxima_part2/local_maxima.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/hands_on/local_maxima_part2/local_maxima.py b/hands_on/local_maxima_part2/local_maxima.py index db89ba3..723b640 100644 --- a/hands_on/local_maxima_part2/local_maxima.py +++ b/hands_on/local_maxima_part2/local_maxima.py @@ -7,4 +7,14 @@ def find_maxima(x): Output: idx -- list of indices of the local maxima in x """ - return [] + sol = [] + for idx, num in enumerate(x): + if idx == 0 and num > x[idx+1]: + sol.append(idx) + elif idx == len(x)-1 and num > x[idx-1]: + sol.append(idx) + else: + if x[idx-1] < num and num > x[idx+1]: + sol.append(idx) + + return sol