我试图找到一对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:索引太多"
答案 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
此函数将返回所提供数组中最近值的索引(这样我们就不会使用全局变量)。请注意,这是在一维数组中搜索,就像您对Y
和tidyr::separate
一样。