Numpy 3d Array Transpose Python