find_peaks does not identify a peak at the start of the array
Unfortunately, find_peaks()
works by comparing neighboring values - so will not identify peaks that occur at the beginning or end of the array. One workaround is to use np.concatenate()
to insert the minimum value of the array at the beginning and end, and then subtract 1 from the peaks variable:
>>> import numpy as np
>>> peaks, _ = find_peaks(np.concatenate(([min(arr1)],arr1,[min(arr1)])), distance=10)
>>> peaks-1
array([ 0, 10, 27], dtype=int64)
def rolling_window(a, window):
shape = a.shape[:-1] + (a.shape[-1] - window + 1, window)
strides = a.strides + (a.strides[-1],)
return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides)
def get_peaks(arr, window):
maxss = np.argmax(rolling_window(arr1, window), axis=1)
return np.where(maxss == 0)[0]
>>> arr1 = np.array([1. , 0.73381293, 0.75649351, 0.77693474, 0.77884614,
0.81055903, 0.81402439, 0.78798586, 0.78839588, 0.82967961,
0.8448 , 0.83276451, 0.82539684, 0.81762916, 0.82722515,
0.82101804, 0.82871127, 0.82825041, 0.82086957, 0.8347826 ,
0.82666665, 0.82352942, 0.81270903, 0.81191224, 0.83180428,
0.84975767, 0.84044236, 0.85057473, 0.8394649 , 0.80000001,
0.83870965, 0.83962262, 0.85039371, 0.83359748, 0.84019768,
0.83281732, 0.83660132])
>>> get_peaks(arr1, 10)
array([ 0, 10, 27])
Credit for rolling window function : Rolling window for 1D arrays in Numpy?