How To Check Dimension In Numpy