How to find row of 2d array in 3d numpy array,"np.all(np.all(test, axis=2), axis=1)"