Getting Array Input From User In Python For Loop