Python Get Dimension Of Numpy Array Size