Inputting An Array In Python Interview