Find the row indexes of several values in a numpy array

Question

I have an array X:

X = np.array([[4,  2],
              [9,  3],
              [8,  5],
              [3,  3],
              [5,  6]])

And

Answer 1

Another way is to use cdist function from scipy.spatial.distance like this:

    np.nonzero(cdist(X, searched_values) == 0)[0]

Basically, we get row numbers of X which have distance zero to a row in searched_values, meaning they are equal. Makes sense if you look on rows as coordinates.