how to get column, row number based on value in matrix from numpy

Question

How to return column and row number based on value in matrix?

import numpy as np

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

something like

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

Answer 1

Use np.argwhere function

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)

Output:

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

Answer 2

Use np.where function.

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