How to find row of 2d array in 3d numpy array,"array([True, False, False, True], dtype=bool)"