Numpy Array Number Of Rows Python Interview Programming