如何在二维数组中查找特定元素？

Question

a2D = np.arraya2D = np.array([[1, 2, 3, 2, 1], [1, 4, 5, 3, 2]])

因此，如果我有一个这样的数组，我不确定是否可以找到数组的特定部分，其中的值大于其邻居，因此如果它发现 3 和 5 是我希望它打印的值/返回数组位置a2D[0][2] , a2D[1][2]

Answer 1

You can try to traverse each element in your a2D array and check if the current element is greater than its neighbours (using 2 for loops):

import numpy as np

a2D = np.arraya2D = np.array([[1, 2, 3, 2, 1], [1, 4, 5, 3, 2]])

for i in range(len(a2D)):
    for j in range(1, len(a2D[i])-1):
        if a2D[i][j] > a2D[i][j-1] and a2D[i][j] > a2D[i][j+1]:
            print('a2D[{}][{}]'.format(i,j))
            break

Output:

a2D[0][2]
a2D[1][2]

I am assuming that such an element is unique, that's why the break . If that is not the case, the break would not be there.