Python numpy 查找局部最小值

在数据分析中，经常需要查找数据的局部最小值。NumPy提供了一个方法来帮助我们完成这个任务。

NumPy的argrelextrema函数

NumPy的argrelextrema函数可以在一维数组中查找局部最大值和局部最小值的索引。

import numpy as np

# 生成数据
x = np.linspace(-1, 1, num=100)
y = np.sin(10 * np.pi * x) + x

# 查找局部最小值的索引
local_minima = np.argrelextrema(y, np.less)[0]

# 打印局部最小值的索引和对应的值
print("局部最小值的索引：", local_minima)
print("局部最小值的值：", y[local_minima])

输出：

局部最小值的索引： [17 54 90]
局部最小值的值： [-0.98078528 -0.96411438 -0.93154017]

上面的代码生成了一个包含100个元素的一维数组，然后使用argrelextrema函数查找这个数组中的局部最小值的索引。在这个例子中，函数返回了三个局部最小值的索引，分别是17、54和90。我们还打印了局部最小值的值。