Найти пики с помощью scipy "find_peaks ()"?

Question

Я хотел бы обнаруживать пики, например, с помощью библиотеки scipy и ее функции find_peaks() с помощью этого простого исходного кода:

import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import find_peaks

ecg = np.load("sample.npy")
peaks, _ = find_peaks(ecg)
plt.plot(ecg)
plt.plot(peaks, ecg[peaks], "x")
plt.show()

и я хотел бы получить что-то вроде этого:

но я хотел бы получить этот результат ↑ для большего количества образцов ... сделайте его более общим.

Answer 1

Возьмите текущую разницу, а затем установите порог этого массива, чтобы получить пики.

import itertools
import operator
import numpy as np

arr = np.array(accumulate(your_array, operator.sub))
arr = np.where[arr > threshold]