How To Take Array Input From The User In Python