Take Array Input From User In Python