我希望能在numpy数组中获取比我的搜索值大的最近值的索引。例如:
我知道我可以通过
findNearestAbove(np.array([0.,1.,1.4,2.]), 1.5)
应该返回3(即2的索引)。我知道我可以通过
np.abs(a-value).argmin()
来获取最近的索引,我发现min(a[np.where(a-value >= 0.)[0]])
返回所需的数组值。因此,np.where(a == min(a[np.where(a-value >= 0.)[0]]))[0]
可能会给我所需的索引。但是,这看起来有些复杂,并且我担心它在多维数组的情况下可能会出错。有没有改进这个方法的建议?