Index confusion in numpy arrays,"A[np.ix_([0, 2], [0, 1], [1, 2])]"