如何在 numpy 数组中找到最接近的值?示例:
np.find_nearest(array, value)
在 NumPy 中,没有像np.find_nearest()查找数组中最近的值这样的直接函数。但是,您可以使用 NumPy 函数组合来实现此功能。以下是如何在 NumPy 数组中找到给定值最近的值:
np.find_nearest()
下面是实现此目的的 Python 函数:
import numpy as np def find_nearest(array, value): array = np.asarray(array) idx = (np.abs(array - value)).argmin() return array[idx] # Example usage: array = np.array([1, 2, 3, 4, 5]) value = 2.5 nearest_value = find_nearest(array, value) nearest_value = find_nearest(array, value) pr print(f"Array: {array}") print(f"Nearest value to {value}: {nearest_value}")
np.asarray(array)
array
np.abs(array - value)
value
.argmin()
array[idx]
idx
通过对绝对差值使用argmin()函数,您可以有效地在 NumPy 数组中找到与任何给定值最接近的值。
argmin()