How to access different rows of a multidimensional NumPy array?