How To Take Array Input From User In Python