在numpy数组中查找最接近的值 [英] Find nearest value in numpy array

问题描述

是否有一种numpy-thonic方式，例如函数，以找到数组中的最近值?

Is there a numpy-thonic way, e.g. function, to find the nearest value in an array?

示例:

np.find_nearest( array, value )

推荐答案

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

array = np.random.random(10)
print(array)
# [ 0.21069679  0.61290182  0.63425412  0.84635244  0.91599191  0.00213826
#   0.17104965  0.56874386  0.57319379  0.28719469]

value = 0.5

print(find_nearest(array, value))
# 0.568743859261

这篇关于在numpy数组中查找最接近的值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！