Find the row indexes of several values in a numpy array,np.where(out.ravel())[0]