python：find_peaks

需求

已知一条曲线(x, y)，想要找到各个peak或dip，以及对应的x值。

应用：已知一幅频谱图，要求找到各个peak及对应的频率

解决方案

1、直接利用已有的方法
scipy.signal.find_peaks

2、自己写一个方法
思路：1.利用sort(list(zip(y,x)))全部排序；2.再根据各个条件筛出结果，比如y值大小，相邻peak的间距等。这样看与方法1思路是可能是相同的。故先直接利用已有方法1。

方案1

直接利用方法：scipy.signal.find_peaks。

方法说明：

 scipy.signal.find_peaks(x, height=None, 
threshold=None, distance=None, prominence=None, width=None, wlen=None, 
rel_height=0.5, plateau_size=None)

输入

• x: 数据
• 其他参数：各种筛选条件。方法将根据这些筛选条件来筛选peak。各输入参数意义详见官方文档

输出：

• peaks： ndarray,找到的peaks的索引id
• properties：dict，包含找到的peaks的各种信息，比如properties[‘peak_heights’]

实例

from scipy.signal import find_peaks
# 已导入需要处理的数据(x,y)
plt.plot(x,y)
plt.xlabel('freq/Hz')
plt.ylabel('amp')

peak_id,peak_property = find_peaks(y, height=2000, distance=20)
peak_freq = x[peak_id]
peak_height = peak_property['peak_heights']
print('peak_freq',peak_freq)
print('peak_height',peak_height)

输出结果：正确。

peak_freq [1.2125e+08 3.0000e+08 4.2125e+08]
peak_height [9277.83035228 4566.30860382 4744.32457053]