How To Get User Input For Numpy Array