Find Index In Array In Python Interview