首页 >后端开发 >Python教程 >如何查找 NumPy 数组中最接近的值?

如何查找 NumPy 数组中最接近的值?

Barbara Streisand
Barbara Streisand原创
2024-11-29 21:00:131088浏览

How to Find the Closest Value in a NumPy Array?

在 Numpy 数组中查找最近值

在 numpy 中,可以通过简单的过程找到给定目标值的最近值。让我们定义一个函数来满足此要求:

import numpy as np

def find_nearest(array, value):
    array = np.asarray(array)
    idx = (np.abs(array - value)).argmin()
    return array[idx]

该函数接受一个数组和一个目标值。它将数组转换为 numpy 数组,并计算数组每个元素与目标值之间的绝对差。使用 argmin() 确定具有最小绝对差的元素的索引。最后,返回该索引处的元素。

用法示例:

让我们用一个示例来说明 find_nearest() 函数的用法:

array = np.random.random(10)
print(array)

print(find_nearest(array, value=0.5))

在此示例中,创建了一个包含 10 个元素的随机数组。然后使用 find_nearest() 函数查找数组中最接近 0.5 的元素。输出将是数组中最接近 0.5 的值。

以上是如何查找 NumPy 数组中最接近的值?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn