擅长:python、mysql、java
<p>这可以通过定义一个作用于一维数组的函数并将其应用于二维数组的行来实现:</p>
<pre><code>e = np.array([[1,2,3], [4,5,6]])
# function to find position of nearest value in 1D array
def find_nearest(a, val):
return np.abs(a - val).argmin()
# apply it
np.apply_along_axis(find_nearest, axis = 1, arr = e, val = 2)
</code></pre>