как получить столбец, номер строки на основе значения в матрице из NumPy

Question

Как вернуть номер столбца и строки на основе значения в матрице?

import numpy as np

matrix = np.random.randint(9, size=(3,3))
#matrix
#[[2 3 3]
# [6 2 4]
# [1 8 4]]

что-то вроде

matrix.where(2,[]) => (0,0), (1,1)
matrix.where(4,[]) => (1,2), (2,2)

Answer 1

Использование np.argwhere функция

import numpy as np

matrix = np.random.randint(9, size=(3,3))
print(matrix)
# array([[2, 1, 3],
#        [7, 2, 4],
#        [1, 7, 7]])

np.argwhere(matrix == 7)

Выход:

array([[1, 0],
       [2, 1],
       [2, 2]], dtype=int64)

Answer 2

Используйте функцию np.where.

matrix = np.random.randint(9, size=(3,3))
np.where(matrix == 2)