Taking Array Input From User In Python