Python - 找到最接近的位置和值

时间:2016-08-15 11:30:06

标签: python numpy indexing

我试图找到一对X和Y的最近点,以便访问其值。在我的例子中,在X方向(np.arrange(0,X.max(),1),我想从Y = 0中提取最接近的值,以在" values array"中获取其值:

import numpy as np
import matplotlib.pyplot as plt
#My coordinates are given here :
X = np.array([0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4 ,5, 0, 1, 2, 3, 4 ,5])
Y = np.array([-2.5, -1.5, 0, 0, 2, 2.5, 2, 1.5, 1, -1, -1.2,-2.5, 0.2, 0.5, 0, -0.5, -0.1,0.05])
plt.scatter(X,Y);plt.show()
#The corresponding values are :
values = np.array([-1.1, -9, 10, 10, 20, 25, 21, 15, 0, 2, -2,-5, 2, 50, 0, -5, -1,5])


# I thought to use a for loop : 

def find_index(x,y):
    xi=np.searchsorted(X,x)
    yi=np.searchsorted(Y,y)
    return xi,yi

 for i in arange(float(0),float(X.max()),1):
      print i
      thisLat, thisLong = find_index(i,0)
      print thisLat, thisLong
      values[thisLat,thisLong]

但是我得到了一个错误:" IndexError:索引太多"

1 个答案:

答案 0 :(得分:4)

您可以使用比import numpy as np def find_nearest(array, value): ''' Find nearest value is an array ''' idx = (np.abs(array-value)).argmin() return idx haystack = np.arange(10) needle = 5.8 idf = find_nearest(haystack, needle) print haystack[idf] # This will return 6 循环更快的内容:

X

此函数将返回所提供数组中最近值的索引(这样我们就不会使用全局变量)。请注意,这是在一维数组中搜索,就像您对Ytidyr::separate一样。