python查找数组元素中最大值_在numpy数组中查找局部最大值

函数^{}中存在一个bulit，它完成了此任务：import numpy as np

from scipy.signal import argrelextrema

a = np.array([1,2,3,4,5,4,3,2,1,2,3,2,1,2,3,4,5,6,5,4,3,2,1])

# determine the indices of the local maxima

maxInd = argrelextrema(a, np.greater)

# get the actual values using these indices

r = a[maxInd] # array([5, 3, 6])

这将为r提供所需的输出。

从SciPy版本1.1开始，您还可以使用find_peaks。下面是从文档本身获取的两个示例。

使用height参数，可以选择高于某个阈值的所有最大值(在本例中，所有非负最大值；如果必须处理噪声基线，这非常有用；如果要找到最小值，只需将输入乘以-1)：import matplotlib.pyplot as plt

from scipy.misc import electrocardiogram

from scipy.signal import find_peaks

import numpy as np

x = electrocardiogram()[2000:4000]

peaks, _ = find_peaks(x, height=0)

plt.plot(x)

plt.plot(peaks, x[peaks], "x")

plt.plot(np.zeros_like(x), "--", color="gray")

plt.show()

另一个非常有用的参数是distance，它定义了两个峰值之间的最小距离：peaks, _ = find_peaks(x, distance=150)

# difference between peaks is >= 150

print(np.diff(peaks))

# prints [186 180 177 171 177 169 167 164 158 162 172]

plt.plot(x)

plt.plot(peaks, x[peaks], "x")

plt.show()