How To Get User Input For Numpy Python