在python中找到两个局部最小值

Question

我有一个t值列表。 我寻找最小值的代码如下：

for i in np.arange(0,499,1):
    if t[i]<t[i-1] and t[i]<t[i+1] :                       
        t_min.append(t[i])

我的t值每次都会更改，因此最小值可能出现在开始或结尾处，在这种情况下此代码将无法工作。 因此，我需要一个适用于t值范围的通用代码。

Answer 1

您可以使用％运算符在末尾循环，并在迭代器的长度上增加一个。 这会将您的数组“视为一个圆”，这正是您真正想要的。

t_min = []
for i in range(len(t)):
    if  t[i] < min(t[i - 1], t[(i + 1) % len(t)]):
        t_min.append(t[i])

编辑：修正我取值的范围，以便第一个元素不会被检查两次，这要感谢@Jasper指出

Answer 2

我建议不要使用scipy.signal.argrelmin来遍历数组，它会找到所有局部最小值。 您可以从中选择最喜欢的两个。

from scipy.signal import argrelmin
import numpy as np
t = np.sin(np.linspace(0, 4*np.pi, 500))
relmin = argrelmin(t)[0]
print(relmin)

输出[187 437] 。

要将数组视为环绕，请使用argrelmin(t, mode='wrap')

---

如果没有回绕，argrelmin不会将数组的开头和结尾识别为局部最小值的候选对象。 （对“局部最小值”有不同的解释：一种允许端点，另一种不允许端点。）如果希望在函数达到最小值时将端点包括在内，请执行以下操作：

if t[0] < t[1]:
    relmin = np.append(relmin, 0)
if t[-1] < t[-2]:
    relmin = np.append(relmin, len(t)-1)

现在输出为[187 437 0] 。