row-wise indexing in numpy,"i = np.array([[0], [1]])"