How To Take User Input For Array In Python