在numpy数组中找到最接近给定值的索引.仅限于外部索引

Question

我的数组称为指示：

[[ 315.    326.31  341.57    0.     18.43   33.69   45.  ]
[ 303.69  315.    333.43    0.     26.57   45.     56.31]
[ 288.43  296.57  315.      0.     45.     63.43   71.57]
[ 270.    270.    270.      0.     90.     90.     90.  ]
[ 251.57  243.43  225.    180.    135.    116.57  108.43]
[ 236.31  225.    206.57  180.    153.43  135.    123.69]
[ 225.    213.69  198.43  180.    161.57  146.31  135.  ]]

我想搜索数组，并从给定值中找到最接近值的索引(例如45(。这就是我到目前为止所做的：

x = np.abs(directions-45)
idx = np.where(x == x.min())

这起作用，在返回所有符合此标准的索引。但是，我想将返回的索引限制在数组外边缘上的那些，即顶部和底部行，最左右的列。如果一个数字更接近不在外边缘中的给定值，那么我想扩展搜索直到找到外边缘的最接近数字。

谢谢！

Answer 1

您可以添加该行

x[1:-1, 1:-1] = x.max()+1

在找到idx之前，要覆盖中心中值的值高于任何边缘值。

x = np.abs(directions-45)
x[1:-1, 1:-1] = x.max()+1
idx = np.where(x == x.min())
idx
Out[25]: (array([0], dtype=int64), array([6], dtype=int64))