Subsetting a 2D numpy array,"a[[1, 2, 3], [1, 2, 3]]"