diff --git a/hands_on/local_maxima/local_maxima.py b/hands_on/local_maxima/local_maxima.py index 1575179..9a64704 100644 --- a/hands_on/local_maxima/local_maxima.py +++ b/hands_on/local_maxima/local_maxima.py @@ -19,4 +19,20 @@ def local_maxima(x): Output: idx -- list of indices of the local maxima in x """ - return [] + max_ls = [] + for i in range(len(x)): + if i ==0: + if x[i] > x[i+1]: + max_ls.append(i) + elif i == len(x)-1: + if x[i] > x[i-1]: + max_ls.append(i) + else: + if x[i] >= x[i+1] and x[i] >= x[i-1]: + max_ls.append(i) + return max_ls + + + + +