Index confusion in numpy arrays,"array([[[1, 2], [4, 5]], [[13, 14], [16, 17]]])"