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