Как найти номер строки и номер столбца определенного элемента c в многомерном массиве, созданном с помощью numpy?

Question

В многомерном массиве, созданном с использованием numpy в python, например,

matrix=np.array[[0,0,0,0,0], [0,0,0,0,1], [0,0,0,0,0], [0,0,0,0,0], [0,0,0,0,0]] 

, если бы мы рассматривали его как матрицу. Есть ли способ найти номер строки и номер столбца элемента '1' или любого другого указанного c элемента внутри матрицы?

Answer 1

Одним из возможных решений является использование np.arg, где :

np.argwhere(matrix == 1)

, но обратите внимание, что ваш массив может содержать несколько элементов только с этим значением, поэтому он возвращает массив 2-D , где каждая строка содержит индексы каждого найденного элемента.

Если вы хотите только первый такой элемент, запустите:

np.argwhere(matrix == 1)[0]

На этот раз вы получите массив 1-D , содержащий 2 элемента (номер строки и столбца).