Python Numpy Number Of Rows In Shapes