How To Save And Load Numpy Array Shape In Python