查找

Question

我使用

index = (np.abs(array - value)).argmin() 

找到具有最小絕對差的值數組索引中的第二最接近的指數值。

但是，有沒有一個很好的乾淨的方式，例如這樣找到第二個最接近索引的值？

Answer 1

我想這樣的作品

a = np.linspace(0,10,30) 
array([ 0.  , 0.34482759, 0.68965517, 1.03448276, 
     1.37931034, 1.72413793, 2.06896552, 2.4137931 , 
     2.75862069, 3.10344828, 3.44827586, 3.79310345, 
     4.13793103, 4.48275862, 4.82758621, 5.17241379, 
     5.51724138, 5.86206897, 6.20689655, 6.55172414, 
     6.89655172, 7.24137931, 7.5862069 , 7.93103448, 
     8.27586207, 8.62068966, 8.96551724, 9.31034483, 
     9.65517241, 10.  ]) 
n = np.pi 
a[np.argsort(np.abs(a-n))[1]] 
# Output 3.4482758620689657 
# the closest value is 3.103... 

Answer 2

你可以得到第k個數組a的最小元素的索引，而無需使用整個數組排序argpartition

np.argpartition(a, k)[k] 

• http://docs.scipy.org/doc/numpy/reference/generated/numpy.argpartition.html