python - 从数据中查找峰值

Question

这个问题已经有答案了:

Peak detection algorithm in Python (4 个回答)

已关闭 5 年前。

我通过 matplotlib 在 x 轴上绘制频率，在 y 轴上绘制强度。

import numpy as np
import matplotlib.pyplot as plt
from scipy.signal import argrelmax, argrelmin

data = np.loadtxt('pr736024.dat')
z=np.array(data[:,1])   
y = (z-z.min())/(z.max()-z.min()) #for normalization

x = np.loadtxt('frq.txt')

plt.plot(x,y, linewidth = 0.2, color='black')
plt.plot(x,indexes, "o")
plt.show()

我想获得峰值处的强度值(图中可以看到 6 个峰值)。我怎样才能实现这个目标？

Answer 1

这个答案改编 self 在 here 中的答案。 .

这里有一个 numpythonic 解决方案(这比显式执行循环要好得多)。

您必须定义一个阈值，在该阈值之上检测到最大值，在您的情况下可以是 0.2。

我使用roll函数将数字+1或-1移动到该位置。此外，“峰值”被定义为局部最大值，其中前一个和后一个数字都小于中心值。

完整代码如下:

import numpy as np
import matplotlib.pyplot as plt

# input signal
x = np.arange(1,100,1)
y = 0.3 * np.sin(t) + 0.7 * np.cos(2 * t) - 0.5 * np.sin(1.2 * t)
threshold = 0.2

# max
maxi = np.where(np.where([(y - np.roll(y,1) > 0) & (y - np.roll(y,-1) > 0)],y, 0)> threshold, y,np.nan)