Index confusion in numpy arrays,"B = numpy.array([A[0, 0, 1], A[2, 1, 2]])"