How To Get Array Input From User In Python