Numpy Shape Function