擅长:python、mysql、java
<p>使用<a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmin.html" rel="nofollow noreferrer">^{<cd1>}</a>可以找到最接近要搜索的元素的索引。这将输出展平向量tough的索引,但您可以使用一些基本数学找到2D:</p>
<pre><code>tosearch = 2
e = np.array([[1, 2, 3], [4, 5, 6]])
idx = np.argmin(np.abs(e - tosearch))
print (idx / e.shape[1], idx % e.shape[1])
</code></pre>