Build Numpy Array Shape In Python