Getting input from user in Numpy One Dimensional Array using while Loop Python