数据分析——scipy.signal.argrelextrema求数组中的极大值和极小值

最新推荐文章于 2024-01-05 09:19:29 发布

原创最新推荐文章于 2024-01-05 09:19:29 发布 · 1.8w 阅读

58 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #极大值 #极小值

数据分析专栏收录该内容

16 篇文章

订阅专栏

本文介绍如何利用Python中的Scipy库函数argrelextrema来寻找一维数据序列中的局部最大值（峰值）和局部最小值（谷值）。通过设置参数order可以调整比较的范围，从而更准确地识别峰谷位置。文中提供了完整的代码示例，并展示了不同order值下峰谷识别的变化。

该文章已生成可运行项目，

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

data_x = np.arange(start = 0, stop = 40, step = 1, dtype='int')
data_y = np.array([98,96,97,100,95,105,75,50,45,42,
					51,85,90,92,91,89,101,62,65,52,
					47,58,55,75,89,92,94,91,89,79,
					85,65,42,55,48,50,85,88,95,100])

# Find peaks
# order:两侧使用多少点进行比较
peak_indexes = signal.argrelextrema(data_y, np.greater, order=1)
peak_indexes = peak_indexes[0]

# Find valleys
# order:两侧使用多少点进行比较
valley_indexes = signal.argrelextrema(data_y, np.less, order=1)
valley_indexes = valley_indexes[0]

(fig, ax) = plt.subplots()

# Plot all data
ax.plot(data_x, data_y)

# Plot peaks
peak_x = peak_indexes
peak_y = data_y[peak_indexes]
ax.scatter(peak_x, peak_y, marker='o', color='red', label="Peaks")

# Plot valleys
valley_x = valley_indexes
valley_y = data_y[valley_indexes]
ax.scatter(valley_x, valley_y, marker='o', color='green', label="Valleys")

# 添加标题
plt.title('Find peaks and valleys using argrelextrema()')
# 添加图例
plt.legend(loc='best')
# 保存图像
plt.savefig('peaks-valleys.png')
# 显示图像
plt.show()