Python 넘파이 배열에서 가장 가까운 값 찾기, Find nearest value in numpy 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]

print(find_nearest(array, value=0.5))
# 0.568743859261