Argrelmin 函数用于计算一维或多维数组的局部极小值的位置,但在使用时需要注意它不会直接返回局部极小值本身,而是返回一个元组,包含每个维度的局部极小值的位置索引。因此需要在函数返回后,通过将这些索引传入原数组的取值操作来获取局部极小值本身。
示例代码:
import numpy as np
a = np.array([1, 2, 3, 2, 1])
min_indices = np.argrelmin(a)
print(min_indices) # 输出:(array([0, 4], dtype=int64),)
min_values = a[min_indices]
print(min_values) # 输出:[1 1]