python if in list index