Numpy Array Number Of Rows Python Snake