Select subset of rows of numpy array based on a selection of rows in another array