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