Find nearest value in numpy array

后端 未结 16 1874
被撕碎了的回忆
被撕碎了的回忆 2020-11-22 10:18

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

Example:

np.find_nearest( array, value )
16条回答
  •  醉梦人生
    2020-11-22 11:01

    Maybe helpful for ndarrays:

    def find_nearest(X, value):
        return X[np.unravel_index(np.argmin(np.abs(X - value)), X.shape)]
    

提交回复
热议问题