Есть ли какая-либо функция / реализация в python, подобная пикам (N) в MATLAB?

Question

Я могу получить 2D матрицу / 3D график, используя peaks(N), где N - любое число в MATLAB. Есть ли способ сделать это в примере python?

MATLAB:

Создать матрицу пиков 5 на 5 и отобразить поверхность:

figure
peaks(5);

Как это сделать в python?

Answer 1

Вы можете использовать SciPy

from scipy.signal import find_peaks
peaks, _ = find_peaks(x, height=0) # x is the signal

print("x-values: ", peaks," y-values: ",  x[peaks])

Документация SciPy для поиска пиков

.. Или быстрое решение, если вы подаете сигнал не слишком шумно, тогда вы можете вручную сгладить сигнал, дифференцировать сглаженный сигнал, найти пороговое значение и сосчитать нули:)