Python Get Dimensions Of Numpy Array Size