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