Numpy Save 3d Array Representation